From 1828425da66bef6251739033139491bfe982d1ca Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 13 Nov 2022 16:50:36 -0600
Subject: [PATCH] Typing fixes for mypy 0.990
---
arraycontext/container/traversal.py | 3 +--
arraycontext/context.py | 1 +
test/test_utils.py | 4 ++--
3 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
index 642aaf1..c41b464 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -681,8 +681,7 @@ def flatten(
# NOTE: we can't do much if the array context fails to ravel,
# since it is the one responsible for the actual memory layout
if hasattr(subary_c, "strides"):
- # Mypy has a point: nobody promised a strides attr.
- strides_msg = f" and strides {subary_c.strides}" # type: ignore[attr-defined] # noqa: E501
+ strides_msg = f" and strides {subary_c.strides}"
else:
strides_msg = ""
diff --git a/arraycontext/context.py b/arraycontext/context.py
index dcc601e..2378550 100644
--- a/arraycontext/context.py
+++ b/arraycontext/context.py
@@ -175,6 +175,7 @@ if TYPE_CHECKING:
# {{{ typing
ScalarLike = Union[int, float, complex, np.generic]
+
SelfType = TypeVar("SelfType")
diff --git a/test/test_utils.py b/test/test_utils.py
index 493b9e7..5edf66b 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -46,7 +46,7 @@ def test_pt_actx_key_stringification_uniqueness():
# {{{ test_dataclass_array_container
-def test_dataclass_array_container():
+def test_dataclass_array_container() -> None:
from typing import Optional
from dataclasses import dataclass, field
from arraycontext import dataclass_array_container
@@ -109,7 +109,7 @@ def test_dataclass_array_container():
# {{{ test_dataclass_container_unions
-def test_dataclass_container_unions():
+def test_dataclass_container_unions() -> None:
from dataclasses import dataclass
from arraycontext import dataclass_array_container
--
GitLab