From 26212c14e5570aff40b90c11495d97dada4272fb Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Fri, 9 Apr 2021 18:09:53 -0400 Subject: [PATCH] Reactivate Megatron tests an use less workers --- .circleci/config.yml | 2 +- tests/test_modeling_megatron_bert.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 9435b90f27..4b490f3259 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -145,7 +145,7 @@ jobs: key: v0.4-torch-{{ checksum "setup.py" }} paths: - '~/.cache/pip' - - run: python -m pytest -n 8 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt + - run: python -m pytest -n 4 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt - store_artifacts: path: ~/transformers/tests_output.txt - store_artifacts: diff --git a/tests/test_modeling_megatron_bert.py b/tests/test_modeling_megatron_bert.py index d592e1e522..5be4716d33 100644 --- a/tests/test_modeling_megatron_bert.py +++ b/tests/test_modeling_megatron_bert.py @@ -265,7 +265,6 @@ class MegatronBertModelTester: @require_torch -@unittest.skip("Temporary skip to make the CI pass reliably.") class MegatronBertModelTest(ModelTesterMixin, unittest.TestCase): all_model_classes = (