diff --git a/tests/integration/tests/test_nvidia_gpu_operator.py b/tests/integration/tests/test_nvidia_gpu_operator.py index 9b3a2eb841..e380f0b074 100644 --- a/tests/integration/tests/test_nvidia_gpu_operator.py +++ b/tests/integration/tests/test_nvidia_gpu_operator.py @@ -58,6 +58,7 @@ def _check_nvidia_drivers_loaded(instance: harness.Instance) -> Mapping[str, boo @pytest.mark.node_count(1) @pytest.mark.tags(tags.WEEKLY) +@pytest.mark.tags(tags.GPU) @pytest.mark.parametrize( "gpu_operator_version", NVIDIA_GPU_OPERATOR_SUPPORTED_UBUNTU_VERSIONS.keys() ) diff --git a/tests/integration/tests/test_util/tags.py b/tests/integration/tests/test_util/tags.py index 49f0204df6..94cbfad5e3 100644 --- a/tests/integration/tests/test_util/tags.py +++ b/tests/integration/tests/test_util/tags.py @@ -6,6 +6,7 @@ PULL_REQUEST = "pull_request" NIGHTLY = "nightly" WEEKLY = "weekly" +GPU = "gpu" TEST_LEVELS = [PULL_REQUEST, NIGHTLY, WEEKLY]