Set num threads before juliacall init

pysr/julia_helpers.py

@@ -10,6 +10,7 @@ if os.environ.get("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes") != "yes":
     )
 
 os.environ["PYTHON_JULIACALL_HANDLE_SIGNALS"] = "yes"
+os.environ["JULIA_NUM_THREADS"] = "auto"
 
 import juliapkg
 from juliacall import Main as jl

pysr/test/test.py

@@ -594,23 +594,6 @@ class TestMiscellaneous(unittest.TestCase):
         with self.assertRaises(ValueError):
             model.fit(X, y)
 
-    def test_changed_options_warning(self):
-        """Check that a warning is given if Julia options are changed."""
-        if julia_helpers.julia_kwargs_at_initialization is None:
-            julia_helpers.init_julia(julia_kwargs={"threads": 2, "optimize": 3})
-
-        cur_init = julia_helpers.julia_kwargs_at_initialization
-
-        threads_to_change = cur_init["threads"] + 1
-        with warnings.catch_warnings():
-            warnings.simplefilter("error")
-            with self.assertRaises(Exception) as context:
-                julia_helpers.init_julia(
-                    julia_kwargs={"threads": threads_to_change, "optimize": 3}
-                )
-            self.assertIn("Julia has already started", str(context.exception))
-            self.assertIn("threads", str(context.exception))
-
     def test_extra_sympy_mappings_undefined(self):
         """extra_sympy_mappings=None errors for custom operators"""
         model = PySRRegressor(unary_operators=["square2(x) = x^2"])