Enable adjustable maxsize

gui/app.py

@@ -16,6 +16,7 @@ os.system("bash install_pysr.sh")
 
 def greet(
     file_obj: tempfile._TemporaryFileWrapper,
+    maxsize: int,
     col_to_fit: str,
     niterations: int,
     binary_operators: list,
@@ -68,6 +69,7 @@ def greet(
     os.system(
         f"python run_pysr_and_save.py "
         f"--niterations {niterations} "
+        f"--maxsize {maxsize} "
         f"--binary_operators '{binary_operators}' "
         f"--unary_operators '{unary_operators}' "
         f"--col_to_fit {col_to_fit} "
@@ -89,7 +91,13 @@ def main():
                 minimum=1,
                 maximum=1000,
                 default=40,
-                label="Number of iterations",
+                label="Number of Iterations",
+            ),
+            gr.inputs.Slider(
+                minimum=7,
+                maximum=35,
+                default=20,
+                label="Maximum Complexity"
             ),
             gr.inputs.CheckboxGroup(
                 choices=["+", "-", "*", "/", "^"],

gui/run_pysr_and_save.py

@@ -21,6 +21,7 @@ empty_df = pd.DataFrame(
 if __name__ == "__main__":
     parser = ArgumentParser()
     parser.add_argument("--niterations", type=int)
+    parser.add_argument("--maxsize", type=int)
     parser.add_argument("--binary_operators", type=str)
     parser.add_argument("--unary_operators", type=str)
     parser.add_argument("--col_to_fit", type=str)
@@ -31,6 +32,7 @@ if __name__ == "__main__":
     unary_operators = eval(args.unary_operators)
     col_to_fit = args.col_to_fit
     filename = args.filename
+    maxsize = args.maxsize
 
     os.environ["PATH"] += ":/home/user/.local/bin/"
 
@@ -52,6 +54,7 @@ if __name__ == "__main__":
         model = PySRRegressor(
             update=False,
             progress=False,
+            maxsize=maxsize,
             niterations=niterations,
             binary_operators=binary_operators,
             unary_operators=unary_operators,