Spaces:
Sleeping
Sleeping
| """Functions for initializing the Julia environment and installing deps.""" | |
| from typing import Any, Callable, Union, cast | |
| import numpy as np | |
| from juliacall import convert as jl_convert # type: ignore | |
| from numpy.typing import NDArray | |
| from .deprecated import init_julia, install | |
| from .julia_import import jl | |
| jl_convert = cast(Callable[[Any, Any], Any], jl_convert) | |
| jl.seval("using Serialization: Serialization") | |
| jl.seval("using PythonCall: PythonCall") | |
| Serialization = jl.Serialization | |
| PythonCall = jl.PythonCall | |
| jl.seval("using SymbolicRegression: plus, sub, mult, div, pow") | |
| 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 _load_cluster_manager(cluster_manager: str): | |
| jl.seval(f"using ClusterManagers: addprocs_{cluster_manager}") | |
| return jl.seval(f"addprocs_{cluster_manager}") | |
| def jl_array(x): | |
| if x is None: | |
| return None | |
| return jl_convert(jl.Array, x) | |
| def jl_serialize(obj: Any) -> NDArray[np.uint8]: | |
| buf = jl.IOBuffer() | |
| Serialization.serialize(buf, obj) | |
| return np.array(jl.take_b(buf)) | |
| def jl_deserialize(s: Union[NDArray[np.uint8], None]): | |
| if s is None: | |
| return s | |
| buf = jl.IOBuffer() | |
| jl.write(buf, jl_array(s)) | |
| jl.seekstart(buf) | |
| return Serialization.deserialize(buf) | |