integration: define 'gpu' tag for Nvidia-reated tests.

Signed-off-by: Nashwan Azhari <[email protected]>
canonical · Jan 8, 2025 · c609ec1 · c609ec1
1 parent a38d1e2
commit c609ec1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -6,6 +6,7 @@
 PULL_REQUEST = "pull_request"
 NIGHTLY = "nightly"
 WEEKLY = "weekly"
+GPU = "gpu"
 
 TEST_LEVELS = [PULL_REQUEST, NIGHTLY, WEEKLY]