tpu

Author: Borda

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

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