mypy

Author: ksagiyam

ufl/algorithms/apply_restrictions.py

@@ -50,8 +50,8 @@ def __init__(
         self.current_restriction: Literal["+", "-"] | None = side
         self.default_restrictions = default_restrictions
         # Caches for propagating the restriction with map_expr_dag
-        self.vcaches = {"+": {}, "-": {}}
-        self.rcaches = {"+": {}, "-": {}}
+        self.vcaches: dict[Literal["+", "-"], dict] = {"+": {}, "-": {}}
+        self.rcaches: dict[Literal["+", "-"], dict] = {"+": {}, "-": {}}
         if self.current_restriction is None:
             self._rp = {
                 "+": RestrictionPropagator("+", default_restrictions),

ufl/domain.py

@@ -384,7 +384,7 @@ def join_domains(domains: Sequence[AbstractDomain], expand_mesh_sequence: bool =
 # TODO: Move these to an analysis module?
 
 
-def extract_domains(expr: Expr | Form, expand_mesh_sequence: bool = True) -> tuple[AbstractDomain]:
+def extract_domains(expr: Expr | Form, expand_mesh_sequence: bool = True) -> tuple[AbstractDomain, ...]:
     """Return all domains expression is defined on.
 
     Args:

ufl/integral.py

@@ -71,7 +71,7 @@ def __init__(
         else:
             self._extra_domain_integral_type_map = {
                 d: extra_domain_integral_type_map[d]
-                for d in sort_domains(extra_domain_integral_type_map.keys())
+                for d in sort_domains(tuple(extra_domain_integral_type_map.keys()))
             }
 
     def reconstruct(