Type-ignore ndarray assignments for numpy 2.2

inducer · Dec 10, 2024 · 4f1ec21 · 4f1ec21
1 parent c21f7ea
commit 4f1ec21
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py
@@ -963,7 +963,7 @@ def map_numpy_array(self,
         import numpy
         result = numpy.empty(expr.shape, dtype=object)
         for i in numpy.ndindex(expr.shape):
-            result[i] = self.rec(expr[i], *args, **kwargs)
+            result[i] = self.rec(expr[i], *args, **kwargs)  # type: ignore[assignment]
 
         # True fact: ndarrays aren't expressions
         return result  # type: ignore[return-value]

diff --git a/pymbolic/mapper/evaluator.py b/pymbolic/mapper/evaluator.py
@@ -163,7 +163,7 @@ def map_numpy_array(self, expr: np.ndarray) -> ResultT:
         import numpy
         result = numpy.empty(expr.shape, dtype=object)
         for i in numpy.ndindex(expr.shape):
-            result[i] = self.rec(expr[i])
+            result[i] = self.rec(expr[i])  # type: ignore[call-overload]
         return result  # type: ignore[return-value]
 
     def map_multivector(self, expr: MultiVector) -> ResultT: