From c4f00b8bb5c104a9557c31ef048cd321d608eca3 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Fri, 13 Dec 2024 11:16:14 +0200
Subject: [PATCH] mypy: fix errors from numpy 2.2
---
arraycontext/container/__init__.py | 3 ++-
arraycontext/loopy.py | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/arraycontext/container/__init__.py b/arraycontext/container/__init__.py
index afe4a40..e70b51d 100644
--- 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
index 7b1d6a0..af57932 100644
--- a/arraycontext/loopy.py
+++ b/arraycontext/loopy.py
@@ -103,7 +103,7 @@ def _get_scalar_func_loopy_program(actx, 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], [
--
GitLab