mypy: fix some new numpy errors

inducer · Dec 15, 2024 · 0957223 · 0957223
1 parent 219a4c4
commit 0957223
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pytential/linalg/skeletonization.py b/pytential/linalg/skeletonization.py
@@ -796,7 +796,8 @@ def skeletonize_by_proxy(
     skels: np.ndarray = np.empty((wrangler.nrows, wrangler.ncols), dtype=object)
     for ibrow in range(wrangler.nrows):
         for ibcol in range(wrangler.ncols):
-            skels[ibrow, ibcol] = _skeletonize_block_by_proxy_with_mats(
+            # NOTE: type annotations for object arrays are not there yet
+            skels[ibrow, ibcol] = _skeletonize_block_by_proxy_with_mats(  # type: ignore[call-overload]
                     actx, ibrow, ibcol, places, proxy, wrangler, tgt_src_index,
                     id_eps=id_eps, id_rank=id_rank,
                     max_particles_in_box=max_particles_in_box)

diff --git a/pytential/qbx/target_assoc.py b/pytential/qbx/target_assoc.py
@@ -132,7 +132,7 @@
 
 class target_status_enum(Enum):  # noqa
     c_name = "TargetStatus"
-    dtype = np.int32
+    dtype = np.dtype(np.int32)
     c_value_prefix = ""
 
     UNMARKED = 0
@@ -142,7 +142,7 @@ class target_status_enum(Enum):  # noqa
 
 class target_flag_enum(Enum):  # noqa
     c_name = "TargetFlag"
-    dtype = np.int32
+    dtype = np.dtype(np.int32)
     c_value_prefix = ""
 
     INTERIOR_OR_EXTERIOR_VOLUME_TARGET = 0