fix fetch tests with parallelism

huggingface · May 1, 2024 · 52b35f7 · 52b35f7
1 parent 3ecf0ec
commit 52b35f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
@@ -172,8 +172,8 @@ def to_dict(self):
             steps.append({"store_artifacts": {"path": "~/transformers/splitted_tests.txt"}})
 
             test_command = ""
-            if self.timeout:
-                test_command = f"timeout {self.timeout} "
+            if self.command_timeout:
+                test_command = f"timeout {self.command_timeout} "
             test_command += f"python3 -m pytest -rsfE -p no:warnings --tb=line -n {self.pytest_num_workers} " + " ".join(pytest_flags)
             test_command += " $(cat splitted_tests.txt)"
         if self.marker is not None: