Spaces:
Sleeping
Sleeping
Clean up default params
Browse files- 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 |
-
|
| 26 |
-
|
| 27 |
-
)
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|