Force bounds checking in Windows tests

MilesCranmer · Dec 10, 2022 · c428709 · c428709
1 parent b4238b4
commit c428709
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/CI_Windows.yml b/.github/workflows/CI_Windows.yml
@@ -57,9 +57,15 @@ jobs:
             python -c 'import pysr; pysr.install()'
       - name: "Run tests"
         run: python -m pysr.test main
+        env:
+          JULIA_CHECK_BOUNDS: yes
       - name: "Install Torch"
         run: pip install torch # (optional import)
       - name: "Run Torch tests"
         run: python -m pysr.test torch
+        env:
+          JULIA_CHECK_BOUNDS: yes
       - name: "Run custom env tests"
         run: python -m pysr.test env
+        env:
+          JULIA_CHECK_BOUNDS: yes
diff --git a/pysr/julia_helpers.py b/pysr/julia_helpers.py
@@ -157,6 +157,9 @@ def init_julia(julia_project=None, quiet=False, julia_kwargs=None):
     if julia_kwargs is None:
         julia_kwargs = {"optimize": 3}
 
+    if "check_bounds" not in julia_kwargs and "JULIA_CHECK_BOUNDS" in os.environ:
+        julia_kwargs["check_bounds"] = os.environ["JULIA_CHECK_BOUNDS"]
+
     from julia.core import JuliaInfo, UnsupportedPythonError
 
     _julia_version_assertion()