fix: external pylib bugs

Author: mtshiba

crates/erg_compiler/context/inquire.rs

@@ -1687,6 +1687,7 @@ impl Context {
                 let mut errs = TyCheckErrors::empty();
                 // method: obj: 1, subr: (self: Int, other: Int) -> Int
                 // non-method: obj: Int, subr: (self: Int, other: Int) -> Int
+                // FIXME: staticmethod
                 let is_method = subr
                     .self_t()
                     .map_or(false, |self_t| self.subtype_of(obj.ref_t(), self_t));

crates/erg_compiler/declare.rs

@@ -49,8 +49,8 @@ impl<A: ASTBuildable> GenericASTLowerer<A> {
             hir::Expr::TypeAsc(tasc) => enum_unwrap!(tasc.expr.as_ref(), hir::Expr::Accessor)
                 .local_name()
                 .map(Str::rc),
-            hir::Expr::Accessor(acc) => acc.var_info().py_name.clone(),
-            _ => sig.inspect().cloned(),
+            hir::Expr::Accessor(hir::Accessor::Ident(ident)) => ident.vi.py_name.clone(),
+            _ => sig.escaped(),
         };
         let found_body_t = chunk.ref_t();
         let ident = match &sig.pat {

crates/erg_compiler/error/tycheck.rs

@@ -805,12 +805,18 @@ passed keyword args:    {kw_args_len}"
             "english" =>sup_type.push_str("supertype: "),
         );
         sup_type.push_str_with_color_and_attr(format!("{sup_t}"), ERR, ATTR);
+        let hint = switch_lang!(
+            "japanese" => "これがあなたの定義した型ならば、型を共変もしくは反変にしてみてください(<: Output T もしくは <: Input T)",
+            "simplified_chinese" => "如果这是您定义的类型，请尝试将类型变为协变或逆变(<: Output T 或 <: Input T)",
+            "traditional_chinese" => "如果這是您定義的類型，請嘗試將類型變為協變或逆變(<: Output T 或 <: Input T)",
+            "english" => "If this is the type you defined, try to make the type covariant or contravariant (<: Output T or <: Input T)",
+        );
         Self::new(
             ErrorCore::new(
                 vec![SubMessage::ambiguous_new(
                     loc,
                     vec![sub_type.to_string(), sup_type.to_string()],
-                    None,
+                    Some(hint.to_string()),
                 )],
                 switch_lang!(
                     "japanese" => "不変な型パラメータを一意に決定できません",

crates/erg_compiler/lib/external/numpy.d/__init__.d.er

@@ -9,15 +9,21 @@
     ndim: Nat
     dtype: Type
     size: Nat
+    copy: |T, S: [Nat; _]|(self: .NDArray(T, S),) -> .NDArray(T, S)
     reshape: |T, Old: [Nat; _], S: [Nat; _]|(
         self: .NDArray(T, Old),
         shape: {S},
     ) -> .NDArray(T, S)
+    sum: |T <: Num|(self: .NDArray(T, _),) -> T
+    take: (|T|(self: .NDArray(T, _), indice: Nat) -> T) \
+        and (|T|(self: .NDArray(T, _), indices: .NDArray(Nat) or [Nat; _]) -> .NDArray(T, _))
+    tobytes: |T|(self: .NDArray(T, _),) -> Bytes
+    tolist: |T|(self: .NDArray(T, _),) -> [T; _]
 
 .nan: Float
 .Nan: Float
 
-.abs: |T|(object: .NDArray(T),) -> .NDArray(T)
+.abs: |T, S: [Nat; _]|(object: .NDArray(T, S),) -> .NDArray(T, S)
 .add: |T, S: [Nat; _]|(object: .NDArray(T, S), other: .NDArray(T, S)) -> .NDArray(T, S)
 .all: |T <: Num|(object: .NDArray(T),) -> Bool
 .any: |T <: Num|(object: .NDArray(T),) -> Bool

crates/erg_compiler/lib/external/tqdm.d/__init__.d.er

@@ -1,7 +1,6 @@
 .Tqdm! = 'tqdm': (T: Type) -> ClassType
 .Tqdm!(T) <: Iterable T
 .Tqdm!(T).
-    # TODO: iterable should be Comparable
     __call__: (
         iterable: Iterable(T),
         desc := Str,
@@ -30,3 +29,5 @@
         delay := Float,
         gui := Bool,
     ) -> .Tqdm!(T)
+
+.tqdm = .Tqdm!.__call__

crates/erg_parser/ast.rs

@@ -4406,6 +4406,18 @@ impl VarPattern {
         }
     }
 
+    pub fn escaped(&self) -> Option<Str> {
+        match self {
+            Self::Ident(ident) => {
+                let inspect = ident.inspect();
+                Some(Str::rc(
+                    inspect.trim_end_matches('!').trim_start_matches('$'),
+                ))
+            }
+            _ => None,
+        }
+    }
+
     // _!(...) = ... is invalid
     pub fn is_procedural(&self) -> bool {
         match self {
@@ -4502,6 +4514,10 @@ impl VarSignature {
         self.pat.inspect()
     }
 
+    pub fn escaped(&self) -> Option<Str> {
+        self.pat.escaped()
+    }
+
     pub const fn vis(&self) -> &VisModifierSpec {
         self.pat.vis()
     }

examples/external.er

@@ -1,9 +1,12 @@
 time = pyimport "time"
 tqdm = pyimport "tqdm"
 j2 = pyimport "jinja2"
+np = pyimport "numpy"
 
 for! tqdm.Tqdm!(0..<100), _ =>
     time.sleep! 0.01
+for! tqdm.tqdm(0..<100), _ =>
+    time.sleep! 0.01
 
 plt = pyimport "matplotlib/pyplot"
 
@@ -14,3 +17,6 @@ plt.show!()
 
 res = j2.Template("Hello {{ name }}!").render(name:="World")
 assert res == "Hello World!"
+
+arr = np.array([1, 2, 3])
+assert arr.sum() == 6