name change

ROCm · Jul 24, 2024 · 6e87c7e · 6e87c7e
1 parent 81191c1
commit 6e87c7e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py
@@ -180,15 +180,15 @@
     IS_ROCM_SYSTEM = False
     IS_CUDA_SYSTEM = False
 
-_run_rocm_tests = parse_flag_from_env("RUN_ROCM_TESTS", default=False)
+_hf_rocm_test = parse_flag_from_env("HF_ROCM_TEST", default=False)
 
 def skipIfRocm(func=None, *, msg="test doesn't currently work on the ROCm stack"):
     def dec_fn(fn):
         reason = f"skipIfRocm: {msg}"
 
         @wraps(fn)
         def wrapper(*args, **kwargs):
-            if _run_rocm_tests:
+            if _hf_rocm_test:
                 pytest.skip(reason)
             else:
                 return fn(*args, **kwargs)