mypy: fix errors from numpy 2.2

illinois-ceesd · Dec 13, 2024 · c4f00b8 · c4f00b8
1 parent a5eaebe
commit c4f00b8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/arraycontext/container/__init__.py b/arraycontext/container/__init__.py
@@ -309,7 +309,8 @@ def _deserialize_ndarray_container(  # type: ignore[misc]
 
     result = type(template)(template.shape, dtype=object)
     for i, subary in serialized:
-        result[i] = subary
+        # FIXME: numpy annotations don't seem to handle object arrays very well
+        result[i] = subary  # type: ignore[call-overload]
 
     return result
 

diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py
@@ -103,7 +103,7 @@ def get(c_name, nargs, naxes):
         from arraycontext.transform_metadata import ElementwiseMapKernelTag
 
         def sub(name: str) -> Variable | Subscript:
-            return Variable(name)[subscript] if subscript else Variable(name)
+            return Subscript(Variable(name), subscript) if subscript else Variable(name)
 
         return make_loopy_program(
                 [domain_bset], [