mps used only if not in github actions

wolearyc · Sep 16, 2024 · cc86093 · cc86093
1 parent 8beec1b
commit cc86093
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/tests/torch/test_gnn.py b/test/tests/torch/test_gnn.py
@@ -118,7 +118,7 @@ def test_gpu(poscar_ref_structure_fixture: ReferenceStructure) -> None:
         device = "cuda"
     # mps backend doesn't work with github runners
     # https://github.com/actions/runner-images/issues/9918
-    elif torch.backends.mps.is_available() and os.getenv("GITHUB_ACTIONS") == "true":
+    elif torch.backends.mps.is_available() and os.getenv("GITHUB_ACTIONS") != "true":
         device = "mps"
     else:
         return