fixed test

CederGroupHub · Sep 25, 2023 · d5ba24b · d5ba24b
1 parent 516c422
commit d5ba24b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_relaxation.py b/tests/test_relaxation.py
@@ -41,7 +41,7 @@ def test_relaxation(algorithm: Literal["legacy", "fast"]):
 
 
 no_cuda = mark.skipif(not torch.cuda.is_available(), reason="No CUDA device")
-no_mps = mark.skipif(not hasattr(torch.backends, "mps"), reason="No MPS device")
+no_mps = mark.skipif(not torch.backends.mps.is_available(), reason="No MPS device")
 
 
 @mark.parametrize(