run full TPU tests

Lightning-AI · Jul 28, 2020 · c5d9d13 · c5d9d13
1 parent 2e6e254
commit c5d9d13
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dockers/tpu-tests/Dockerfile b/dockers/tpu-tests/Dockerfile
@@ -22,8 +22,8 @@ RUN git clone https://github.com/PyTorchLightning/pytorch-lightning.git && \
 # the source code where the tests live.
 RUN \
     # drop Horovod
-    #python -c "fname = 'pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" && \
-    #pip install --requirement pytorch-lightning/requirements/extra.txt --no-cache-dir && \
+    python -c "fname = 'pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" && \
+    pip install --requirement pytorch-lightning/requirements/extra.txt --no-cache-dir && \
     if [ $TEST_IMAGE -eq 1 ] ; then \
         pip install --requirement pytorch-lightning/requirements/test.txt --no-cache-dir ; \
     else \

diff --git a/dockers/tpu-tests/tpu_test_cases.jsonnet b/dockers/tpu-tests/tpu_test_cases.jsonnet
@@ -21,7 +21,7 @@ local tputests = base.BaseTest {
   command: utils.scriptCommand(
     |||
       cd pytorch-lightning
-      coverage run --source=pytorch_lightning -m pytest tests/models/test_tpu.py -v
+      coverage run --source=pytorch_lightning -m pytest pytorch_lightning tests -v
       test_exit_code=$?
       echo "\n||| END PYTEST LOGS |||\n"
       coverage xml