tpu

Lightning-AI · Jan 26, 2021 · 5d46913 · 5d46913
1 parent bea6694
commit 5d46913
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/dockers/tpu-tests/tpu_test_cases.jsonnet b/dockers/tpu-tests/tpu_test_cases.jsonnet
@@ -21,7 +21,10 @@ local tputests = base.BaseTest {
   command: utils.scriptCommand(
     |||
       cd pytorch-lightning
-      coverage run --source=pytorch_lightning -m pytest tests/models/test_tpu.py tests/backends/test_tpu_backend.py pytorch_lightning/utilities/xla_device_utils.py -v
+      coverage run --source=pytorch_lightning -m pytest -v \
+          pytorch_lightning/utilities/xla_device_utils.py \
+          tests/accelerators/legacy/test_tpu_backend.py \
+          tests/models/test_tpu.py
       test_exit_code=$?
       echo "\n||| END PYTEST LOGS |||\n"
       coverage xml

diff --git a/pyproject.toml b/pyproject.toml
@@ -38,3 +38,10 @@ profile = "black"
 line_length = 120
 force_sort_within_sections = "False"
 order_by_type = "False"
+
+
+[tool.vulture]
+exclude = ["tests"]
+min_confidence = 80
+paths = ["pytorch_lightning"]
+sort_by_size = true