Standard names for JuliaCall env variables

pysr/julia_import.py

@@ -9,7 +9,7 @@ if "juliacall" in sys.modules:
     warnings.warn(
         "juliacall module already imported. "
         "Make sure that you have set the environment variable `PYTHON_JULIACALL_HANDLE_SIGNALS=yes` to avoid segfaults. "
-        "Also note that PySR will not be able to configure `JULIA_NUM_THREADS` or `JULIA_OPTIMIZE` for you."
+        "Also note that PySR will not be able to configure `PYTHON_JULIACALL_THREADS` or `PYTHON_JULIACALL_OPTLEVEL` for you."
     )
 else:
     # Required to avoid segfaults (https://juliapy.github.io/PythonCall.jl/dev/faq/)
@@ -19,9 +19,9 @@ else:
             + "You will experience segfaults if running with multithreading."
         )
 
-    if os.environ.get("JULIA_NUM_THREADS", "auto") != "auto":
+    if os.environ.get("PYTHON_JULIACALL_THREADS", "auto") != "auto":
         warnings.warn(
-            "JULIA_NUM_THREADS environment variable is set to something other than 'auto', "
+            "PYTHON_JULIACALL_THREADS environment variable is set to something other than 'auto', "
             "so PySR was not able to set it. You may wish to set it to `'auto'` for full use "
             "of your CPU."
         )
@@ -29,8 +29,8 @@ else:
     # TODO: Remove these when juliapkg lets you specify this
     for k, default in (
         ("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes"),
-        ("JULIA_NUM_THREADS", "auto"),
-        ("JULIA_OPTIMIZE", "3"),
+        ("PYTHON_JULIACALL_THREADS", "auto"),
+        ("PYTHON_JULIACALL_OPTLEVEL", "3"),
     ):
         os.environ[k] = os.environ.get(k, default)
 

pysr/test/test_startup.py

@@ -106,8 +106,8 @@ class TestStartup(unittest.TestCase):
                 msg="PYTHON_JULIACALL_HANDLE_SIGNALS environment variable is set",
             ),
             dict(
-                code='import os; os.environ["JULIA_NUM_THREADS"] = "1"; import pysr',
-                msg="JULIA_NUM_THREADS environment variable is set",
+                code='import os; os.environ["PYTHON_JULIACALL_THREADS"] = "1"; import pysr',
+                msg="PYTHON_JULIACALL_THREADS environment variable is set",
             ),
             dict(
                 code="import juliacall; import pysr",