MilesCranmer commited on
Commit
a9e19e6
1 Parent(s): 801ce9c

Install Bumper as well

Browse files
Files changed (2) hide show
  1. gui/Dockerfile +3 -2
  2. gui/app.py +1 -0
gui/Dockerfile CHANGED
@@ -32,8 +32,9 @@ RUN python -m venv /home/user/.venv
32
  # Install Python dependencies in a virtual environment
33
  RUN /home/user/.venv/bin/python -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
34
 
35
- # Install Julia dependencies
36
- RUN /home/user/.venv/bin/python -c "import pysr"
 
37
 
38
  WORKDIR $HOME/app
39
  COPY --chown=user . $HOME/app
 
32
  # Install Python dependencies in a virtual environment
33
  RUN /home/user/.venv/bin/python -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
34
 
35
+ # Install and pre-compile Julia dependencies,
36
+ # including the Bumper extension
37
+ RUN /home/user/.venv/bin/python -c "import pysr; pysr.PySRRegressor(bumper=True, verbosity=0, progress=False, max_evals=1).fit([[1]], [1])"
38
 
39
  WORKDIR $HOME/app
40
  COPY --chown=user . $HOME/app
gui/app.py CHANGED
@@ -75,6 +75,7 @@ def greet(
75
 
76
  model = pysr.PySRRegressor(
77
  progress=False,
 
78
  maxsize=maxsize,
79
  niterations=niterations,
80
  binary_operators=binary_operators,
 
75
 
76
  model = pysr.PySRRegressor(
77
  progress=False,
78
+ bumper=True,
79
  maxsize=maxsize,
80
  niterations=niterations,
81
  binary_operators=binary_operators,