Spaces:
Sleeping
Sleeping
Fix precompilation logic
Browse files- pysr/julia_helpers.py +4 -9
pysr/julia_helpers.py
CHANGED
|
@@ -78,17 +78,15 @@ def install(julia_project=None, quiet=False, precompile=None): # pragma: no cov
|
|
| 78 |
processed_julia_project, is_shared = _process_julia_project(julia_project)
|
| 79 |
_set_julia_project_env(processed_julia_project, is_shared)
|
| 80 |
|
| 81 |
-
if
|
| 82 |
os.environ["JULIA_PKG_PRECOMPILE_AUTO"] = "0"
|
| 83 |
|
| 84 |
julia.install(quiet=quiet)
|
| 85 |
Main, init_log = init_julia(julia_project, quiet=quiet, return_aux=True)
|
| 86 |
io_arg = _get_io_arg(quiet)
|
| 87 |
|
| 88 |
-
if precompile is None
|
| 89 |
-
precompile =
|
| 90 |
-
else:
|
| 91 |
-
precompile = True
|
| 92 |
|
| 93 |
if not precompile:
|
| 94 |
Main.eval('ENV["JULIA_PKG_PRECOMPILE_AUTO"] = 0')
|
|
@@ -100,10 +98,7 @@ def install(julia_project=None, quiet=False, precompile=None): # pragma: no cov
|
|
| 100 |
Main.eval("using Pkg")
|
| 101 |
Main.eval(f"Pkg.instantiate({io_arg})")
|
| 102 |
|
| 103 |
-
if precompile
|
| 104 |
-
"JULIA_PKG_PRECOMPILE_AUTO" not in os.environ
|
| 105 |
-
or str(os.environ["JULIA_PKG_PRECOMPILE_AUTO"]) != "0"
|
| 106 |
-
):
|
| 107 |
Main.eval(f"Pkg.precompile({io_arg})")
|
| 108 |
|
| 109 |
if not quiet:
|
|
|
|
| 78 |
processed_julia_project, is_shared = _process_julia_project(julia_project)
|
| 79 |
_set_julia_project_env(processed_julia_project, is_shared)
|
| 80 |
|
| 81 |
+
if precompile == False:
|
| 82 |
os.environ["JULIA_PKG_PRECOMPILE_AUTO"] = "0"
|
| 83 |
|
| 84 |
julia.install(quiet=quiet)
|
| 85 |
Main, init_log = init_julia(julia_project, quiet=quiet, return_aux=True)
|
| 86 |
io_arg = _get_io_arg(quiet)
|
| 87 |
|
| 88 |
+
if precompile is None:
|
| 89 |
+
precompile = init_log["compiled_modules"]
|
|
|
|
|
|
|
| 90 |
|
| 91 |
if not precompile:
|
| 92 |
Main.eval('ENV["JULIA_PKG_PRECOMPILE_AUTO"] = 0')
|
|
|
|
| 98 |
Main.eval("using Pkg")
|
| 99 |
Main.eval(f"Pkg.instantiate({io_arg})")
|
| 100 |
|
| 101 |
+
if precompile:
|
|
|
|
|
|
|
|
|
|
| 102 |
Main.eval(f"Pkg.precompile({io_arg})")
|
| 103 |
|
| 104 |
if not quiet:
|