Pass GITHUB_* env variables via tox

brainglobe · May 30, 2024 · 9dc8bdd · 9dc8bdd
1 parent 9b1b050
commit 9dc8bdd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -140,6 +140,7 @@ omit = benchmarks/*
 [testenv]
 extras =
     dev
+passenv = GITHUB_*
 deps =
     coredev: git+https://github.com/brainglobe/cellfinder.git
 commands =

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -18,7 +18,10 @@ def set_device_arm_macos_ci():
     GitHub runners. This is to avoid the following error:
     https://discuss.pytorch.org/t/mps-back-end-out-of-memory-on-github-action/189773/5
     """
-    if os.getenv("CI") == "true" and torch.backends.mps.is_available():
+    if (
+        os.getenv("GITHUB_ACTIONS") == "true"
+        and torch.backends.mps.is_available()
+    ):
         import keras
 
         keras.src.backend.common.global_state.set_global_attribute(