From 56c281083b254d3f193e35287417aa8f904b349f Mon Sep 17 00:00:00 2001 From: Thibaut Durand Date: Sat, 7 Oct 2023 20:28:52 -0700 Subject: [PATCH] Update dependencies and workflow --- src/coola/utils/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/coola/utils/tensor.py b/src/coola/utils/tensor.py index de112f1e..c9a1ab4d 100644 --- 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