Fixup: get rid of "type alias" since we don't check anyway

GridTools · Feb 25, 2025 · de4f84f · de4f84f
1 parent 16564a6
commit de4f84f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/tests/cartesian_tests/definitions.py b/tests/cartesian_tests/definitions.py
@@ -14,19 +14,14 @@
     cp = None
 
 import datetime
-from typing import List
-
 import numpy as np
 import pytest
 
 from gt4py import cartesian as gt4pyc
 from gt4py.cartesian import utils as gt_utils
 
 
-ParameterSet = type(pytest.param())
-
-
-def _backend_name_as_param(name: str) -> ParameterSet:
+def _backend_name_as_param(name: str):
     marks = []
     if gt4pyc.backend.from_name(name).storage_info["device"] == "gpu":
         marks.append(pytest.mark.requires_gpu)
@@ -38,7 +33,7 @@ def _backend_name_as_param(name: str) -> ParameterSet:
 _ALL_BACKEND_NAMES = list(gt4pyc.backend.REGISTRY.keys())
 
 
-def _get_backends_with_storage_info(storage_info_kind: str) -> List[ParameterSet]:
+def _get_backends_with_storage_info(storage_info_kind: str):
     res = []
     for name in _ALL_BACKEND_NAMES:
         backend = gt4pyc.backend.from_name(name)

diff --git a/tests/cartesian_tests/integration_tests/multi_feature_tests/test_code_generation.py b/tests/cartesian_tests/integration_tests/multi_feature_tests/test_code_generation.py
@@ -30,7 +30,6 @@
 from cartesian_tests.definitions import (
     ALL_BACKENDS,
     CPU_BACKENDS,
-    ParameterSet,
     get_array_library,
 )
 from cartesian_tests.integration_tests.multi_feature_tests.stencil_definitions import (
@@ -769,7 +768,7 @@ def test(
     assert (out_arr[:, :, :] == 388.0).all()
 
 
-def _xfail_dace_backends(param: ParameterSet) -> ParameterSet:
+def _xfail_dace_backends(param):
     if param.values[0].startswith("dace:"):
         marks = param.marks + [
             pytest.mark.xfail(