Spaces:
Running
Running
| """Functions for initializing the Julia environment and installing deps.""" | |
| import warnings | |
| import juliacall | |
| import juliapkg | |
| jl = juliacall.newmodule("PySR") | |
| juliainfo = None | |
| julia_initialized = False | |
| julia_kwargs_at_initialization = None | |
| julia_activated_env = None | |
| def _get_io_arg(quiet): | |
| io = "devnull" if quiet else "stderr" | |
| return f"io={io}" | |
| def _escape_filename(filename): | |
| """Turn a path into a string with correctly escaped backslashes.""" | |
| str_repr = str(filename) | |
| str_repr = str_repr.replace("\\", "\\\\") | |
| return str_repr | |
| def _backend_version_assertion(): | |
| backend_version = jl.seval("string(SymbolicRegression.PACKAGE_VERSION)") | |
| expected_backend_version = juliapkg.status(target="SymbolicRegression").version | |
| if backend_version != expected_backend_version: # pragma: no cover | |
| warnings.warn( | |
| f"PySR backend (SymbolicRegression.jl) version {backend_version} " | |
| f"does not match expected version {expected_backend_version}. " | |
| "Things may break. " | |
| ) | |
| def _load_cluster_manager(cluster_manager): | |
| jl.seval(f"using ClusterManagers: addprocs_{cluster_manager}") | |
| return jl.seval(f"addprocs_{cluster_manager}") | |