fix some pyright errors (#367)

Author: Roger-luo

src/kirin/decl/scan_fields.py

@@ -101,7 +101,9 @@ def _get_field(self, name: str, typ):
         return f
 
     def _post_init_field(self, f: Field, guess: type | None):
-        if isinstance(f, ArgumentField) and is_subhint(guess, tuple[ir.SSAValue, ...]):
+        if isinstance(f, ArgumentField) and (
+            guess and is_subhint(guess, tuple[ir.SSAValue, ...])
+        ):
             f.group = True
         # try to narrow the type based on the guess
         elif isinstance(f, AttributeField):

src/kirin/dialects/ilist/runtime.py

@@ -11,7 +11,7 @@
 
 
 @dataclass
-class IList(ir.Data[Sequence[T]], Generic[T, L]):
+class IList(ir.Data[Sequence[T]], Sequence[T], Generic[T, L]):
     """A simple immutable list."""
 
     data: Sequence[T]
@@ -63,22 +63,25 @@ def __iter__(self):
         return iter(self.data)
 
     @overload
-    def __getitem__(self, index: slice) -> "IList[T, Any]": ...
+    def __getitem__(self, index: int) -> T: ...
 
     @overload
-    def __getitem__(self, index: int) -> T: ...
+    def __getitem__(self, index: slice) -> "IList[T, Any]": ...
 
     def __getitem__(self, index: int | slice) -> T | "IList[T, Any]":
         if isinstance(index, slice):
             return IList(self.data[index])
         return self.data[index]
 
+    def __contains__(self, item: object) -> bool:
+        return item in self.data
+
     def __eq__(self, value: object) -> bool:
         if not isinstance(value, IList):
             return False
         return self.data == value.data
 
-    def unwrap(self) -> "IList[T, L]":
+    def unwrap(self) -> Sequence[T]:
         return self
 
     def print_impl(self, printer: Printer) -> None:

src/kirin/dialects/py/assign.py

@@ -154,6 +154,8 @@ def lower_AugAssign(
                 rhs = ast.Attribute(obj, attr, ast.Load())
             case ast.Subscript(obj, slice, ast.Store()):
                 rhs = ast.Subscript(obj, slice, ast.Load())
+            case _:
+                raise lowering.BuildError(f"unsupported target {node.target}")
         self.assign_item_value(
             state,
             node.target,

src/kirin/ir/traits/callable.py

@@ -43,9 +43,9 @@ def get_signature(cls, stmt: "Statement"):
     def set_signature(cls, stmt: "Statement", signature: "Signature"):
         stmt.attributes["signature"] = signature
 
-    def verify(self, stmt: "Statement"):
+    def verify(self, node: "Statement"):
         from kirin.dialects.func.attrs import Signature
 
-        signature = self.get_signature(stmt)
+        signature = self.get_signature(node)
         if not isinstance(signature, Signature):
             raise ValueError(f"{signature} is not a Signature attribute")

src/kirin/ir/traits/symbol.py

@@ -25,10 +25,10 @@ def get_sym_name(self, stmt: Statement) -> PyAttr[str]:
             raise ValueError(f"Statement {stmt.name} does not have a symbol name")
         return sym_name
 
-    def verify(self, stmt: Statement):
+    def verify(self, node: Statement):
         from kirin.types import String
 
-        sym_name = self.get_sym_name(stmt)
+        sym_name = self.get_sym_name(node)
         if not (isinstance(sym_name, PyAttr) and sym_name.type.is_subseteq(String)):
             raise ValueError(f"Symbol name {sym_name} is not a string attribute")
 
@@ -43,17 +43,17 @@ class SymbolTable(StmtTrait):
     def walk(stmt: Statement):
         return stmt.regions[0].blocks[0].stmts
 
-    def verify(self, stmt: Statement):
-        if len(stmt.regions) != 1:
+    def verify(self, node: Statement):
+        if len(node.regions) != 1:
             raise ValidationError(
-                stmt,
-                f"Statement {stmt.name} with SymbolTable trait must have exactly one region",
+                node,
+                f"Statement {node.name} with SymbolTable trait must have exactly one region",
             )
 
-        if len(stmt.regions[0].blocks) != 1:
+        if len(node.regions[0].blocks) != 1:
             raise ValidationError(
-                stmt,
-                f"Statement {stmt.name} with SymbolTable trait must have exactly one block",
+                node,
+                f"Statement {node.name} with SymbolTable trait must have exactly one block",
             )
 
         # TODO: check uniqueness of symbol names

src/kirin/lowering/python/traits.py

@@ -130,7 +130,7 @@ def lower(
         args, kwargs = self.lower_Call_inputs(stmt, state, node)
         return state.current_frame.push(stmt(*args.values(), **kwargs))
 
-    def verify(self, node: StatementType):
+    def verify(self, node: ir.Statement):
         assert len(node.regions) == 0, "FromPythonCall statements cannot have regions"
         assert (
             len(node.successors) == 0
@@ -264,13 +264,13 @@ def lower(
             state.current_frame.defs[result.name] = result
         return
 
-    def verify(self, stmt: ir.Statement):
+    def verify(self, node: ir.Statement):
         assert (
-            len(stmt.regions) == 1
+            len(node.regions) == 1
         ), "FromPythonWithSingleItem statements must have one region"
         assert (
-            len(stmt.successors) == 0
+            len(node.successors) == 0
         ), "FromPythonWithSingleItem statements cannot have successors"
         assert (
-            len(stmt.results) <= 1
+            len(node.results) <= 1
         ), "FromPythonWithSingleItem statements can have at most one result"