diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py index 6b992612e8..f4682b8933 100644 --- a/examples/pytorch/test_pytorch_examples.py +++ b/examples/pytorch/test_pytorch_examples.py @@ -19,7 +19,6 @@ import json import logging import os import sys -import unittest from unittest.mock import patch import torch @@ -240,7 +239,6 @@ class ExamplesTests(TestCasePlus): self.assertGreaterEqual(result["eval_accuracy"], 0.75) self.assertLess(result["eval_loss"], 0.5) - @unittest.skip("Broken, fix me Sourab") def test_run_squad(self): tmp_dir = self.get_auto_remove_tmp_dir() testargs = f""" diff --git a/utils/tests_fetcher.py b/utils/tests_fetcher.py index d8373da5ef..0480e44c3e 100644 --- a/utils/tests_fetcher.py +++ b/utils/tests_fetcher.py @@ -852,10 +852,10 @@ if __name__ == "__main__": if commit_flags["test_all"]: with open(args.output_file, "w", encoding="utf-8") as f: - if args.filters is None: - f.write("./tests/") - else: - f.write(" ".join(args.filters)) + f.write("tests") + example_file = Path(args.output_file).parent / "examples_test_list.txt" + with open(example_file, "w", encoding="utf-8") as f: + f.write("all") test_files_to_run = get_all_tests() create_json_map(test_files_to_run, args.json_output_file)