From c609ec10b9d7f6ef37451b8fe08b573e8e3cf661 Mon Sep 17 00:00:00 2001 From: Benjamin Schimke Date: Wed, 18 Dec 2024 11:55:41 +0200 Subject: [PATCH] integration: define 'gpu' tag for Nvidia-reated tests. Signed-off-by: Nashwan Azhari --- tests/integration/tests/test_nvidia_gpu_operator.py | 1 + tests/integration/tests/test_util/tags.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/integration/tests/test_nvidia_gpu_operator.py b/tests/integration/tests/test_nvidia_gpu_operator.py index 9b3a2eb84..e380f0b07 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 6aef07a68..f5fa3119d 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]