test_video_llava: use OH_TEST_DEVICE

Signed-off-by: U. Artie Eoff <[email protected]>
uartie · Feb 18, 2025 · 6a166f4 · 6a166f4
1 parent eb4bdc2
commit 6a166f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_video_llava.py b/tests/test_video_llava.py
@@ -8,9 +8,9 @@
 import pytest
 
 from .test_examples import TIME_PERF_FACTOR
+from .utils import OH_TEST_DEVICE
 
-
-if os.environ.get("GAUDI2_CI", "0") == "1":
+if OH_TEST_DEVICE in ["gaudi2"]:
     # Gaudi2 CI baselines
     MODELS_TO_TEST = {
         "bf16": [