Fixes for new typing failures

Author: inducer

arraycontext/impl/pytato/__init__.py

@@ -653,7 +653,7 @@ def _to_frozen(
                 transformed_dag = self.transform_dag(normalized_expr)
 
                 from pytato.tags import PrefixNamed
-                name_hint_tags = []
+                name_hint_tags: list[PrefixNamed] = []
                 for subary in key_to_pt_arrays.values():
                     name_hint_tags.extend(subary.tags_of_type(PrefixNamed))
 

arraycontext/pytest.py

@@ -33,7 +33,7 @@
 THE SOFTWARE.
 """
 
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
 
 from typing_extensions import override
 
@@ -43,6 +43,8 @@
 if TYPE_CHECKING:
     from collections.abc import Callable, Sequence
 
+    import pyopencl as cl
+
     from arraycontext.context import ArrayContext
 
 
@@ -62,14 +64,16 @@ class PytestPyOpenCLArrayContextFactory(PytestArrayContextFactory):
     .. automethod:: __init__
     .. automethod:: __call__
     """
+    device: cl.Device
 
-    def __init__(self, device):
+    def __init__(self, device: cl.Device):
         """
         :arg device: a :class:`pyopencl.Device`.
         """
         self.device = device
 
     @classmethod
+    @override
     def is_available(cls) -> bool:
         try:
             import pyopencl  # noqa: F401
@@ -384,7 +388,10 @@ def inner(metafunc):
             if pyopencl_factories:
                 for arg_dict in arg_values:
                     arg_values_with_actx.extend([
-                        {factory_arg_name: factory(arg_dict["device"]), **arg_dict}
+                        {factory_arg_name: cast(
+                                "type[PytestPyOpenCLArrayContextFactory]",
+                                factory)(cast("cl.Device", arg_dict["device"])),
+                            **arg_dict}
                         for factory in pyopencl_factories
                         ])
 

doc/conf.py

@@ -46,3 +46,12 @@
     ["py:class", r"np.integer"],
     ["py:class", r".*\|.*"],
     ]
+
+sphinxconfig_missing_reference_aliases = {
+    # pyopencl
+    "cl.Device": "obj:pyopencl.Device",
+}
+
+
+def setup(app):
+    app.connect("missing-reference", process_autodoc_missing_reference)  # noqa: F821