3567977611

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/MilesCranmer/PySR/commit/<a class=hub.com/MilesCranmer/PySR/commit/c4a06fd34ad6ceeea926214d721daa8f9164abd0">c4a06fd34<a href="https://github.com/MilesCranmer/PySR/commit/c4a06fd34ad6ceeea926214d721daa8f9164abd0">">Merge </a><a class="double-link" href="https://github.com/MilesCranmer/PySR/commit/<a class="double-link" href="https://github.com/MilesCranmer/PySR/commit/48fd3fbe02b707ee4d118d0666ca20185c21421e">48fd3fbe0</a>">48fd3fbe0</a><a href="https://github.com/MilesCranmer/PySR/commit/c4a06fd34ad6ceeea926214d721daa8f9164abd0"> into 31ffc0ec3">31ffc0ec3</a>

Pull Request Pull Request #230: Package compiler

Run Details

35 of 56 new or added lines in 5 files covered. (62.5%)

916 of 1037 relevant lines covered (88.33%)

0.88 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

51.85

/pysr/package_compiler.py

"""Functions to create a sysimage for PySR."""

from pathlib import Path
import warnings

import numpy as np
from julia.api import JuliaError

from .version import __symbolic_regression_jl_version__
from .julia_helpers import (
    init_julia,
    _julia_version_assertion,
    _set_julia_project_env,
    _get_io_arg,
    _process_julia_project,
    _import_error,
)


def _add_sr_to_julia_project(Main, io_arg):
    Main.eval("using Pkg")
    Main.sr_spec = Main.PackageSpec(
        name="SymbolicRegression",
        url="https://github.com/MilesCranmer/SymbolicRegression.jl",
        rev="v" + __symbolic_regression_jl_version__,
    )
    Main.clustermanagers_spec = Main.PackageSpec(
        name="ClusterManagers",
        rev="v0.4.2",
    )
    Main.packagecompiler_spec = Main.PackageSpec(
        name="PackageCompiler",
        rev="v2.1.0",
    )
    Main.eval(
        "Pkg.add(["
        + ", ".join(["sr_spec", "clustermanagers_spec", "packagecompiler_spec"])
        + f"], {io_arg})"
    )


def _update_julia_project(Main, is_shared, io_arg):
    try:
        if is_shared:
            _add_sr_to_julia_project(Main, io_arg)
        Main.eval("using Pkg")
        Main.eval(f"Pkg.resolve({io_arg})")
    except (JuliaError, RuntimeError) as e:
        raise ImportError(_import_error()) from e


def install(julia_project=None, quiet=False):  # pragma: no cover
    """
    Install PyCall.jl and all required dependencies for SymbolicRegression.jl.

    Also updates the local Julia registry.
    """
    import julia

    _julia_version_assertion()
    # Set JULIA_PROJECT so that we install in the pysr environment
    processed_julia_project, is_shared = _process_julia_project(julia_project)
    _set_julia_project_env(processed_julia_project, is_shared)

    julia.install(quiet=quiet)
    Main = init_julia(julia_project, quiet=quiet)
    io_arg = _get_io_arg(quiet)

    if is_shared:
        # Install SymbolicRegression.jl:
        _add_sr_to_julia_project(Main, io_arg)

    Main.eval("using Pkg")
    Main.eval(f"Pkg.instantiate({io_arg})")
    Main.eval(f"Pkg.precompile({io_arg})")

    if not quiet:
        warnings.warn(
            "It is recommended to restart Python after installing PySR's dependencies,"
            " so that the Julia environment is properly initialized."
        )


def compile(
    julia_project=None,
    quiet=False,
    sysimage_name="pysr.so",
):
    """Create a PackageCompiler.jl sysimage for SymbolicRegression.jl."""
    Main = init_julia(julia_project=julia_project, quiet=quiet)
    cur_project_dir = Main.eval("dirname(Base.active_project())")
    sysimage_path = str(Path(cur_project_dir) / sysimage_name)
    from julia import PackageCompiler

    Main.eval("using SymbolicRegression")

    PackageCompiler.create_sysimage(["SymbolicRegression"], sysimage_path=sysimage_path)

1	"""Functions to create a sysimage for PySR."""
2
3	from pathlib import Path	1✔
4	import warnings	1✔
5
6	import numpy as np	1✔
7	from julia.api import JuliaError	1✔
8
9	from .version import __symbolic_regression_jl_version__	1✔
10	from .julia_helpers import (	1✔
11	init_julia,
12	_julia_version_assertion,
13	_set_julia_project_env,
14	_get_io_arg,
15	_process_julia_project,
16	_import_error,
17	)
18
19
20	def _add_sr_to_julia_project(Main, io_arg):	1✔
21	Main.eval("using Pkg")	1✔
22	Main.sr_spec = Main.PackageSpec(	1✔
23	name="SymbolicRegression",
24	url="https://github.com/MilesCranmer/SymbolicRegression.jl",
25	rev="v" + __symbolic_regression_jl_version__,
26	)
27	Main.clustermanagers_spec = Main.PackageSpec(	1✔
28	name="ClusterManagers",
29	rev="v0.4.2",
30	)
31	Main.packagecompiler_spec = Main.PackageSpec(	1✔
32	name="PackageCompiler",
33	rev="v2.1.0",
34	)
35	Main.eval(	1✔
36	"Pkg.add(["
37	+ ", ".join(["sr_spec", "clustermanagers_spec", "packagecompiler_spec"])
38	+ f"], {io_arg})"
39	)
40
41
42	def _update_julia_project(Main, is_shared, io_arg):	1✔
NEW 43	try:	×
NEW 44	if is_shared:	×
NEW 45	_add_sr_to_julia_project(Main, io_arg)	×
NEW 46	Main.eval("using Pkg")	×
NEW 47	Main.eval(f"Pkg.resolve({io_arg})")	×
NEW 48	except (JuliaError, RuntimeError) as e:	×
NEW 49	raise ImportError(_import_error()) from e	×
50
51
52	def install(julia_project=None, quiet=False): # pragma: no cover
53	"""
54	Install PyCall.jl and all required dependencies for SymbolicRegression.jl.
55
56	Also updates the local Julia registry.
57	"""
58	import julia
59
60	_julia_version_assertion()
61	# Set JULIA_PROJECT so that we install in the pysr environment
62	processed_julia_project, is_shared = _process_julia_project(julia_project)
63	_set_julia_project_env(processed_julia_project, is_shared)
64
65	julia.install(quiet=quiet)
66	Main = init_julia(julia_project, quiet=quiet)
67	io_arg = _get_io_arg(quiet)
68
69	if is_shared:
70	# Install SymbolicRegression.jl:
71	_add_sr_to_julia_project(Main, io_arg)
72
73	Main.eval("using Pkg")
74	Main.eval(f"Pkg.instantiate({io_arg})")
75	Main.eval(f"Pkg.precompile({io_arg})")
76
77	if not quiet:
78	warnings.warn(
79	"It is recommended to restart Python after installing PySR's dependencies,"
80	" so that the Julia environment is properly initialized."
81	)
82
83
84	def compile(	1✔
85	julia_project=None,
86	quiet=False,
87	sysimage_name="pysr.so",
88	):
89	"""Create a PackageCompiler.jl sysimage for SymbolicRegression.jl."""
NEW 90	Main = init_julia(julia_project=julia_project, quiet=quiet)	×
NEW 91	cur_project_dir = Main.eval("dirname(Base.active_project())")	×
NEW 92	sysimage_path = str(Path(cur_project_dir) / sysimage_name)	×
NEW 93	from julia import PackageCompiler	×
94
NEW 95	Main.eval("using SymbolicRegression")	×
96
NEW 97	PackageCompiler.create_sysimage(["SymbolicRegression"], sysimage_path=sysimage_path)	×

MilesCranmer / PySR / 3567977611

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous