Added CLI to run pysr.install() to install Julia dependencies

.github/workflows/CI.yml

@@ -54,7 +54,7 @@ jobs:
             python -m pip install --upgrade pip
             pip install -r requirements.txt
             python setup.py install
-            python -c 'import pysr; pysr.install()'
+            python -m pysr install
       - name: "Install Coverage tool"
         run: pip install coverage coveralls
       - name: "Run tests"
@@ -112,7 +112,7 @@ jobs:
       - name: "Install PySR"
         run: |
             python3 -m pip install .
-            python3 -c 'import pysr; pysr.install()'
+            python3 -m pysr install
       - name: "Run tests"
         run: cd /tmp && python -m pysr.test main
 

.github/workflows/CI_Windows.yml

@@ -54,7 +54,7 @@ jobs:
             python -m pip install --upgrade pip
             pip install -r requirements.txt
             python setup.py install
-            python -c 'import pysr; pysr.install()'
+            python -m pysr install
       - name: "Run tests"
         run: python -m pysr.test main
       - name: "Install Torch"

.github/workflows/CI_large_nightly.yml

@@ -42,7 +42,7 @@ jobs:
             python -m pip install --upgrade pip
             pip install -r requirements.txt
             python setup.py install
-            python -c 'import pysr; pysr.install()'
+            python -m pysr install
       - name: "Run tests"
         run: python -m pysr.test main
       - name: "Run new env test"

.github/workflows/CI_mac.yml

@@ -54,7 +54,7 @@ jobs:
             python -m pip install --upgrade pip
             pip install -r requirements.txt
             python setup.py install
-            python -c 'import pysr; pysr.install()'
+            python -m pysr install
       - name: "Run tests"
         run: python -m pysr.test main
       - name: "Install JAX"

Dockerfile

@@ -28,7 +28,7 @@ ADD ./pysr/ /pysr/pysr/
 RUN pip3 install .
 
 # Install Julia pre-requisites:
-RUN python3 -c 'import pysr; pysr.install()'
+RUN python3 -m pysr install
 
 # metainformation
 LABEL org.opencontainers.image.authors = "Miles Cranmer"

README.md

@@ -83,7 +83,7 @@ pip install -U pysr
 ```
 3. Finally, to install Julia dependencies:
 ```bash
-python3 -c 'import pysr; pysr.install()'
+python3 -m pysr install
 ```
 
 ---

environment.yml

@@ -9,4 +9,5 @@ dependencies:
   - setuptools
   - pyjulia
   - openlibm
-  - openspecfun
+  - openspecfun
+  - click

pysr/__main__.py

@@ -0,0 +1,4 @@
+from pysr._cli.main import cli as _cli
+
+if __name__ == '__main__':
+    _cli(prog_name="cli")

pysr/_cli/main.py

@@ -0,0 +1,16 @@
+import click
+from ..julia_helpers import install
+
+
+@click.group("cli")
+@click.pass_context
+def cli(context):
+    ctx = context
+
+
+@cli.command("install", help="Install required Julia dependencies")
+@click.option("--julia_project", default=None)
+@click.option("--quiet", default=False)
+@click.option("--precompile", default=None)
+def _install(julia_project, quiet, precompile):
+    install(julia_project, quiet, precompile)

pysr/julia_helpers.py

@@ -110,10 +110,8 @@ def install(julia_project=None, quiet=False, precompile=None):  # pragma: no cov
 
 def _import_error():
     return """
-    Required dependencies are not installed or built.  Run the following code in the Python REPL:
-
-        >>> import pysr
-        >>> pysr.install()
+    Required dependencies are not installed or built.  Run the following command:
+        python -m pysr install
     """
 
 
@@ -286,14 +284,14 @@ def _backend_version_assertion(Main):
                 f"does not match expected version {expected_backend_version}. "
                 "Things may break. "
                 "Please update your PySR installation with "
-                "`python -c 'import pysr; pysr.install()'`."
+                "`python -m pysr install`."
             )
     except JuliaError:  # pragma: no cover
         warnings.warn(
             "You seem to have an outdated version of SymbolicRegression.jl. "
             "Things may break. "
             "Please update your PySR installation with "
-            "`python -c 'import pysr; pysr.install()'`."
+            "`python -m pysr install`."
         )
 
 

requirements.txt

@@ -3,3 +3,5 @@ pandas
 numpy
 scikit_learn >= 1.0.0
 julia >= 0.6.0
+click>=8.0.4
+setuptools>=66.0.0