skip loading t5-base in CI (#5371)

* skip loading t5-base in CI * increase timeout
allenai · Aug 20, 2021 · 8370cfa · 8370cfa
1 parent 13de38d
commit 8370cfa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -75,7 +75,7 @@ jobs:
   checks:
     name: ${{ matrix.task.name }}
     runs-on: ${{ matrix.task.runs_on }}
-    timeout-minutes: 20
+    timeout-minutes: 30
     strategy:
       fail-fast: false
       matrix:

diff --git a/tests/modules/transformer/t5_test.py b/tests/modules/transformer/t5_test.py
@@ -3,16 +3,15 @@
 
 from allennlp.modules.transformer.t5 import T5
 from allennlp.nn.parallel import FairScaleFsdpAccelerator
-from allennlp.common.testing import requires_gpu, run_distributed_test, requires_multi_gpu
+from allennlp.common.testing import run_distributed_test, requires_multi_gpu
 
 
-# Mark this as GPU so it runs on a self-hosted runner, which will be a lot faster.
-@requires_gpu
+@pytest.mark.skip("takes too long in CI")
 @pytest.mark.parametrize(
     "pretrained_model_name",
     [
         "t5-base",
-        #  "t5-large",  # Takes too long in CI
+        #  "t5-large",  # Takes WAY too long in CI
     ],
 )
 def test_create_t5_from_pretrained(pretrained_model_name: str):