refactor: use modern-style type hints

src/gt4py/cartesian/backend/base.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import abc
 import copy
 import hashlib
@@ -43,7 +45,7 @@
 REGISTRY = gt_utils.Registry()
 
 
-def from_name(name: str) -> Optional[Type["Backend"]]:
+def from_name(name: str) -> Optional[Type[Backend]]:
     backend = REGISTRY.get(name, None)
     if not backend:
         raise NotImplementedError(
@@ -52,7 +54,7 @@ def from_name(name: str) -> Optional[Type["Backend"]]:
     return backend
 
 
-def register(backend_cls: Type["Backend"]) -> Type["Backend"]:
+def register(backend_cls: Type[Backend]) -> Type[Backend]:
     assert issubclass(backend_cls, Backend) and backend_cls.name is not None
 
     if isinstance(backend_cls.name, str):
@@ -101,9 +103,9 @@ class Backend(abc.ABC):
     #   "disable-code-generation": bool
     #   "disable-cache-validation": bool
 
-    builder: "StencilBuilder"
+    builder: StencilBuilder
 
-    def __init__(self, builder: "StencilBuilder"):
+    def __init__(self, builder: StencilBuilder):
         self.builder = builder
 
     @classmethod
@@ -120,7 +122,7 @@ def filter_options_for_id(
         return filtered_options
 
     @abc.abstractmethod
-    def load(self) -> Optional[Type["StencilObject"]]:
+    def load(self) -> Optional[Type[StencilObject]]:
         """
         Load the stencil class from the generated python module.
 
@@ -135,7 +137,7 @@ def load(self) -> Optional[Type["StencilObject"]]:
         pass
 
     @abc.abstractmethod
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         """
         Generate the stencil class from GTScript's internal representation.
 
@@ -152,7 +154,7 @@ def generate(self) -> Type["StencilObject"]:
 
     @property
     def extra_cache_info(self) -> Dict[str, Any]:
-        """Provide additional data to be stored in cache info file (sublass hook)."""
+        """Provide additional data to be stored in cache info file (subclass hook)."""
         return {}
 
     @property
@@ -240,9 +242,9 @@ def generate_bindings(self, language_name: str) -> Dict[str, Union[str, Dict]]:
 
 
 class BaseBackend(Backend):
-    MODULE_GENERATOR_CLASS: ClassVar[Type["BaseModuleGenerator"]]
+    MODULE_GENERATOR_CLASS: ClassVar[Type[BaseModuleGenerator]]
 
-    def load(self) -> Optional[Type["StencilObject"]]:
+    def load(self) -> Optional[Type[StencilObject]]:
         build_info = self.builder.options.build_info
         if build_info is not None:
             start_time = time.perf_counter()
@@ -263,11 +265,11 @@ def load(self) -> Optional[Type["StencilObject"]]:
 
         return stencil_class
 
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         self.check_options(self.builder.options)
         return self.make_module()
 
-    def _load(self) -> Type["StencilObject"]:
+    def _load(self) -> Type[StencilObject]:
         stencil_class_name = self.builder.class_name
         file_name = str(self.builder.module_path)
         stencil_module = gt_utils.make_module_from_file(stencil_class_name, file_name)
@@ -289,7 +291,7 @@ def check_options(self, options: gt_definitions.BuildOptions) -> None:
                 stacklevel=2,
             )
 
-    def make_module(self, **kwargs: Any) -> Type["StencilObject"]:
+    def make_module(self, **kwargs: Any) -> Type[StencilObject]:
         build_info = self.builder.options.build_info
         if build_info is not None:
             start_time = time.perf_counter()
@@ -323,7 +325,7 @@ def __call__(self, *, args_data: Optional[ModuleData] = None, **kwargs: Any) ->
 class PurePythonBackendCLIMixin(CLIBackendMixin):
     """Mixin for CLI support for backends deriving from BaseBackend."""
 
-    builder: "StencilBuilder"
+    builder: StencilBuilder
 
     #: stencil python source generator method:
     #:  In order to use this mixin, the backend class must implement
@@ -378,7 +380,7 @@ def extra_cache_validation_keys(self) -> List[str]:
         return keys
 
     @abc.abstractmethod
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         pass
 
     def build_extension_module(
@@ -436,7 +438,7 @@ def disabled(message: str, *, enabled_env_var: str) -> Callable[[Type[Backend]],
     else:
 
         def _decorator(cls: Type[Backend]) -> Type[Backend]:
-            def _no_generate(obj) -> Type["StencilObject"]:
+            def _no_generate(obj) -> Type[StencilObject]:
                 raise NotImplementedError(
                     f"Disabled '{cls.name}' backend: 'f{message}'\n",
                     f"You can still enable the backend by hand using the environment variable '{enabled_env_var}=1'",

src/gt4py/cartesian/backend/cuda_backend.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Type
 
 from gt4py import storage as gt_storage
@@ -141,7 +143,7 @@ class CudaBackend(BaseGTBackend, CLIBackendMixin):
     def generate_extension(self, **kwargs: Any) -> Tuple[str, str]:
         return self.make_extension(stencil_ir=self.builder.gtir, uses_cuda=True)
 
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         self.check_options(self.builder.options)
 
         pyext_module_name: Optional[str]

src/gt4py/cartesian/backend/dace_backend.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import copy
 import os
 import pathlib
@@ -302,7 +304,7 @@ def freeze_origin_domain_sdfg(inner_sdfg, arg_names, field_info, *, origin, doma
         for node in states[0].nodes():
             state.remove_node(node)
 
-    # make sure that symbols are passed throught o inner sdfg
+    # make sure that symbols are passed through to inner sdfg
     for symbol in nsdfg.sdfg.free_symbols:
         if symbol not in wrapper_sdfg.symbols:
             wrapper_sdfg.add_symbol(symbol, nsdfg.sdfg.symbols[symbol])
@@ -531,7 +533,7 @@ def keep_line(line):
         return generated_code
 
     @classmethod
-    def apply(cls, stencil_ir: gtir.Stencil, builder: "StencilBuilder", sdfg: dace.SDFG):
+    def apply(cls, stencil_ir: gtir.Stencil, builder: StencilBuilder, sdfg: dace.SDFG):
         self = cls()
         with dace.config.temporary_config():
             # To prevent conflict with 3rd party usage of DaCe config always make sure that any
@@ -765,7 +767,7 @@ class BaseDaceBackend(BaseGTBackend, CLIBackendMixin):
     GT_BACKEND_T = "dace"
     PYEXT_GENERATOR_CLASS = DaCeExtGenerator  # type: ignore
 
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         self.check_options(self.builder.options)
 
         pyext_module_name: Optional[str]

src/gt4py/cartesian/backend/dace_lazy_stencil.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Set, Tuple
 
 import dace
@@ -22,7 +24,7 @@
 
 
 class DaCeLazyStencil(LazyStencil, SDFGConvertible):
-    def __init__(self, builder: "StencilBuilder"):
+    def __init__(self, builder: StencilBuilder):
         if "dace" not in builder.backend.name:
             raise ValueError("Trying to build a DaCeLazyStencil for non-dace backend.")
         super().__init__(builder=builder)

src/gt4py/cartesian/backend/dace_stencil_object.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import copy
 import inspect
 import os
@@ -58,7 +60,7 @@ def add_optional_fields(
 
 @dataclass(frozen=True)
 class DaCeFrozenStencil(FrozenStencil, SDFGConvertible):
-    stencil_object: "DaCeStencilObject"
+    stencil_object: DaCeStencilObject
     origin: Dict[str, Tuple[int, ...]]
     domain: Tuple[int, ...]
     sdfg: dace.SDFG
@@ -95,7 +97,7 @@ def _get_domain_origin_key(domain, origin):
         return domain, origins_tuple
 
     def freeze(
-        self: "DaCeStencilObject", *, origin: Dict[str, Tuple[int, ...]], domain: Tuple[int, ...]
+        self: DaCeStencilObject, *, origin: Dict[str, Tuple[int, ...]], domain: Tuple[int, ...]
     ) -> DaCeFrozenStencil:
         key = DaCeStencilObject._get_domain_origin_key(domain, origin)
         if key in self._frozen_cache:
@@ -135,8 +137,8 @@ def closure_resolver(
         self,
         constant_args: Dict[str, Any],
         given_args: Set[str],
-        parent_closure: Optional["dace.frontend.python.common.SDFGClosure"] = None,
-    ) -> "dace.frontend.python.common.SDFGClosure":
+        parent_closure: Optional[dace.frontend.python.common.SDFGClosure] = None,
+    ) -> dace.frontend.python.common.SDFGClosure:
         return dace.frontend.python.common.SDFGClosure()
 
     def __sdfg__(self, *args, **kwargs) -> dace.SDFG:

src/gt4py/cartesian/backend/gtc_common.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import abc
 import os
 import textwrap
@@ -134,7 +136,7 @@ def __init__(self):
         self.pyext_file_path = None
 
     def __call__(
-        self, args_data: ModuleData, builder: Optional["StencilBuilder"] = None, **kwargs: Any
+        self, args_data: ModuleData, builder: Optional[StencilBuilder] = None, **kwargs: Any
     ) -> str:
         self.pyext_module_name = kwargs["pyext_module_name"]
         self.pyext_file_path = kwargs["pyext_file_path"]
@@ -229,7 +231,7 @@ class BaseGTBackend(gt_backend.BasePyExtBackend, gt_backend.CLIBackendMixin):
     PYEXT_GENERATOR_CLASS: Type[BackendCodegen]
 
     @abc.abstractmethod
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         pass
 
     def generate_computation(self) -> Dict[str, Union[str, Dict]]:

src/gt4py/cartesian/backend/gtcpp_backend.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Type
 
 from gt4py import storage as gt_storage
@@ -129,7 +131,7 @@ class GTBaseBackend(BaseGTBackend, CLIBackendMixin):
     def _generate_extension(self, uses_cuda: bool) -> Tuple[str, str]:
         return self.make_extension(stencil_ir=self.builder.gtir, uses_cuda=uses_cuda)
 
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         self.check_options(self.builder.options)
 
         pyext_module_name: Optional[str]

src/gt4py/cartesian/backend/module_generator.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import abc
 import numbers
 import sys
@@ -121,11 +123,11 @@ class BaseModuleGenerator(abc.ABC):
 
     TEMPLATE_RESOURCE = "stencil_module.py.in"
 
-    _builder: Optional["StencilBuilder"]
+    _builder: Optional[StencilBuilder]
     args_data: ModuleData
     template: jinja2.Template
 
-    def __init__(self, builder: Optional["StencilBuilder"] = None):
+    def __init__(self, builder: Optional[StencilBuilder] = None):
         self._builder = builder
         self.args_data = ModuleData()
         self.template = jinja2.Template(
@@ -135,7 +137,7 @@ def __init__(self, builder: Optional["StencilBuilder"] = None):
         )
 
     def __call__(
-        self, args_data: ModuleData, builder: Optional["StencilBuilder"] = None, **kwargs: Any
+        self, args_data: ModuleData, builder: Optional[StencilBuilder] = None, **kwargs: Any
     ) -> str:
         """
         Generate source code for a Python module containing a StencilObject.
@@ -176,7 +178,7 @@ def __call__(
         return module_source
 
     @property
-    def builder(self) -> "StencilBuilder":
+    def builder(self) -> StencilBuilder:
         """
         Expose the builder reference.
 
@@ -205,7 +207,7 @@ def generate_class_name(self) -> str:
         """
         Generate the name of the stencil class.
 
-        This should ususally be deferred to the chosen caching strategy via
+        This should usually be deferred to the chosen caching strategy via
         the builder object (see default implementation).
         """
         return self.builder.class_name

src/gt4py/cartesian/backend/numpy_backend.py

@@ -6,6 +6,8 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from __future__ import annotations
+
 import pathlib
 from typing import TYPE_CHECKING, Any, ClassVar, Dict, Type, Union, cast
 
@@ -51,7 +53,7 @@ def generate_implementation(self) -> str:
         return f"computation.run({', '.join(params)})"
 
     @property
-    def backend(self) -> "NumpyBackend":
+    def backend(self) -> NumpyBackend:
         return cast(NumpyBackend, self.builder.backend)
 
 
@@ -98,7 +100,7 @@ def generate_bindings(self, language_name: str) -> Dict[str, Union[str, Dict]]:
         super().generate_bindings(language_name)
         return {self.builder.module_path.name: self.make_module_source()}
 
-    def generate(self) -> Type["StencilObject"]:
+    def generate(self) -> Type[StencilObject]:
         self.check_options(self.builder.options)
         src_dir = self.builder.module_path.parent
         if not self.builder.options._impl_opts.get("disable-code-generation", False):