MilesCranmer commited on
Commit
f335ea1
1 Parent(s): 4a5693e

Clean up default params

Browse files
Files changed (1) hide show
  1. pysr/julia_helpers.py +8 -4
pysr/julia_helpers.py CHANGED
@@ -22,10 +22,14 @@ if os.environ.get("JULIA_NUM_THREADS", "auto") != "auto":
22
  "of your CPU."
23
  )
24
 
25
- os.environ["PYTHON_JULIACALL_HANDLE_SIGNALS"] = os.environ.get(
26
- "PYTHON_JULIACALL_HANDLE_SIGNALS", "yes"
27
- )
28
- os.environ["JULIA_NUM_THREADS"] = os.environ.get("JULIA_NUM_THREADS", "auto")
 
 
 
 
29
 
30
  import juliapkg
31
  from juliacall import Main as jl
 
22
  "of your CPU."
23
  )
24
 
25
+ # TODO: Remove these when juliapkg lets you specify this
26
+ for k, default in (
27
+ ("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes"),
28
+ ("JULIA_NUM_THREADS", "auto"),
29
+ ("JULIA_OPTIMIZE", "3"),
30
+ ):
31
+ os.environ[k] = os.environ.get(k, default)
32
+
33
 
34
  import juliapkg
35
  from juliacall import Main as jl