Fix environ assertion

MilesCranmer · Jan 26, 2024 · d325c42 · d325c42
1 parent 4037c2d
commit d325c42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/julia_helpers.py b/pysr/julia_helpers.py
@@ -3,7 +3,7 @@
 import warnings
 
 # Required to avoid segfaults (https://juliapy.github.io/PythonCall.jl/dev/faq/)
-if os.environ["PYTHON_JULIACALL_HANDLE_SIGNALS"] not in {"yes", ""}:
+if os.environ.get("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes") != "yes":
     warnings.warn(
         "PYTHON_JULIACALL_HANDLE_SIGNALS environment variable is set to something other than 'yes' or ''. "
         + "You will experience segfaults if running with multithreading."