Make sure tests can run locally when GITHUB_ACTIONS env variable is m…

…issing2
brainglobe · May 23, 2024 · d4b7ecd · d4b7ecd
1 parent e98e0af
commit d4b7ecd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/conftest.py b/tests/core/conftest.py
@@ -21,7 +21,7 @@ def set_device_arm_macos_ci():
     https://discuss.pytorch.org/t/mps-back-end-out-of-memory-on-github-action/189773/5
     """
     if (
-        os.environ["GITHUB_ACTIONS"] == "true"
+        os.getenv("GITHUB_ACTIONS") == "true"
         and torch.backends.mps.is_available()
     ):
         keras.src.backend.common.global_state.set_global_attribute(