Fix nightly tests (#1696)

* Debug start * Fix * Workflow
huggingface · Jul 11, 2023 · d0a7991 · d0a7991
1 parent 180ef7c
commit d0a7991
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -194,7 +194,9 @@ def test_cross_validation(self):
         """.split()
         with mock.patch.dict(os.environ, {"TESTING_MOCKED_DATALOADERS": "0"}):
             output = run_command(self._launch_args + testargs, return_stdout=True)
-            results = ast.literal_eval(re.findall("({.+})", output)[-1])
+            results = re.findall("({.+})", output)
+            results = [r for r in results if "accuracy" in r][-1]
+            results = ast.literal_eval(results)
             self.assertGreaterEqual(results["accuracy"], 0.75)
 
     def test_multi_process_metrics(self):