Fix all is_torch_tpu_available issues (#17936)

* Fix all is_torch_tpu_available
2022-06-29 11:03:33 -04:00
parent 77b76672e2
commit 7c4c6f6084
10 changed files with 22 additions and 19 deletions
--- a/src/transformers/testing_utils.py
+++ b/src/transformers/testing_utils.py
@@ -467,7 +467,7 @@ def require_torch_tpu(test_case):
    """
    Decorator marking a test that requires a TPU (in PyTorch).
    """
-    return unittest.skipUnless(is_torch_tpu_available(), "test requires PyTorch TPU")(test_case)
+    return unittest.skipUnless(is_torch_tpu_available(check_device=False), "test requires PyTorch TPU")(test_case)


 if is_torch_available():