Fix the CI

Summary: https://github.com/pytorch/pytorch/pull/17995 's CI has verified it should fix the CI.

Reviewed By: bddppq

Differential Revision: D14447674

fbshipit-source-id: 50085db9ae7421b5be216ed0a2216234babfdf6c

diff --git a/test/test_torch.py b/test/test_torch.py
index 02171df..88f230b 100644 (file)
--- a/test/test_torch.py
+++ b/test/test_torch.py
@@ -4468,7 +4468,7 @@ class _TestTorchMixin(object):
 
     def test_linspace(self):
         devices = ['cpu'] if not torch.cuda.is_available() else ['cpu', 'cuda']
-        for _device in devices:
+        for device in devices:
             _from = random.random()
             to = _from + random.random()
             res1 = torch.linspace(_from, to, 137, device=device)