diff --git a/arraycontext/pytest.py b/arraycontext/pytest.py index 0d534c95bb5dbd1e0f1af86d34a16870bffd74f0..f2dedae4eacb5767b184c02b30a5bb0b7f918da2 100644 --- a/arraycontext/pytest.py +++ b/arraycontext/pytest.py @@ -35,6 +35,7 @@ from arraycontext.impl.pyopencl import PyOpenCLArrayContext from arraycontext.impl.pytato import PytatoArrayContext import pyopencl as cl from pyopencl.tools import _ContextFactory +from typing import List class _PyOpenCLArrayContextFactory(_ContextFactory): @@ -109,7 +110,7 @@ def pytest_generate_tests_for_array_contexts(metafunc, actx_list=None) -> None: for arg_dict in arg_values: dev = arg_dict["device"] - extra_factories = [] + extra_factories: List[str] = [] for factory in actx_factories: if "actx_factory" in arg_dict: