From d5ba24b9f265954bfb712a42e1795543b49af27e Mon Sep 17 00:00:00 2001 From: BowenD-UCB <84425382+BowenD-UCB@users.noreply.github.com> Date: Mon, 25 Sep 2023 14:57:57 -0700 Subject: [PATCH] fixed test --- tests/test_relaxation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_relaxation.py b/tests/test_relaxation.py index abe78617..adfbe35d 100644 --- 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(