Update dependencies and workflow

durandtibo · Oct 8, 2023 · 56c2810 · 56c2810
1 parent 835e25e
commit 56c2810
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coola/utils/tensor.py b/src/coola/utils/tensor.py
@@ -73,7 +73,7 @@ def is_mps_available() -> bool:
     if not is_torch_available():
         return False
     try:
-        torch.ones(1, device=torch.device("mps"))
+        torch.ones(1, device="mps")
         return True
     except RuntimeError:
         return False