[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

docs/gen_param_docs.py

@@ -5,6 +5,7 @@
 placeholder with auto-generated parameter docs from param_groupings.yml
 and PySRRegressor docstrings.
 """
+
 import re
 import sys
 from pathlib import Path

examples/pysr_demo.ipynb

@@ -770,12 +770,10 @@
    "source": [
     "from pysr import jl\n",
     "\n",
-    "jl.seval(\n",
-    "    \"\"\"\n",
+    "jl.seval(\"\"\"\n",
     "import Pkg\n",
     "Pkg.add(\"Primes\")\n",
-    "\"\"\"\n",
-    ")"
+    "\"\"\")"
    ]
   },
   {
@@ -834,17 +832,15 @@
    },
    "outputs": [],
    "source": [
-    "jl.seval(\n",
-    "    \"\"\"\n",
+    "jl.seval(\"\"\"\n",
     "function p(i::T) where T\n",
     "    if 0.5 < i < 1000\n",
     "        return T(prime(round(Int, i)))\n",
     "    else\n",
     "        return T(NaN)\n",
     "    end\n",
     "end\n",
-    "\"\"\"\n",
-    ")"
+    "\"\"\")"
    ]
   },
   {

pysr/expression_specs.py

@@ -275,18 +275,15 @@ def _template_macro_str(self):
             template_inputs.append(
                 f"parameters=({', '.join([f'{p}={self.parameters[p]}' for p in self.parameters]) + ','})"
             )
-        return dedent(
-            f"""
+        return dedent(f"""
         @template_spec({', '.join(template_inputs) + ','}) do {', '.join(self.variable_names)}
             {self.combine}
         end
-        """
-        )
+        """)
 
     def julia_expression_options(self):
         f_combine = jl.seval(self.combine)
-        creator = jl.seval(
-            """
+        creator = jl.seval("""
         function _pysr_create_template_structure(
             @nospecialize(function_symbols::AbstractVector),
             @nospecialize(combine::Function),
@@ -301,8 +298,7 @@ def julia_expression_options(self):
             structure = SymbolicRegression.TemplateStructure{tuple_symbol}(combine, num_features)
             return (; structure)
         end
-        """
-        )
+        """)
         return creator(self.function_symbols, f_combine, self.num_features)
 
     @property
@@ -327,8 +323,7 @@ def parametric_expression_deprecation_warning(
 ):
     function_name = "f"
     var_names = list(variable_names)
-    message = dedent(
-        f"""
+    message = dedent(f"""
         ParametricExpressionSpec is deprecated. you should switch to TemplateExpressionSpec with explicit parameters indexed by category.
 
         Since you have `max_parameters={max_parameters}` and `variable_names=[{", ".join(f'"{v}"' for v in var_names)}]`, you could migrate like this:
@@ -343,8 +338,7 @@ def parametric_expression_deprecation_warning(
             X = np.column_stack([X, category])       # add the category column
 
         Finally, do not pass `category` when calling .fit().
-    """
-    ).strip()
+    """).strip()
     wrapped = "\n".join(textwrap.fill(line, 88) for line in message.splitlines())
     warnings.warn(wrapped, FutureWarning, stacklevel=3)
 
@@ -416,7 +410,7 @@ def _search_output_to_callable_expressions(
     equations: pd.DataFrame, search_output, i: int | None
 ) -> pd.DataFrame:
     equations = copy.deepcopy(equations)
-    (_, all_out_hof) = search_output
+    _, all_out_hof = search_output
     out_hof = all_out_hof[i] if i is not None else all_out_hof
     expressions = []
     callables = []

pysr/logger_specs.py

@@ -48,17 +48,15 @@ class TensorBoardLoggerSpec(AbstractLoggerSpec):
 
     def create_logger(self) -> AnyValue:
         # We assume that TensorBoardLogger is already imported via `julia_extensions.py`
-        make_logger = jl.seval(
-            """
+        make_logger = jl.seval("""
             function make_logger(log_dir::AbstractString, overwrite::Bool, log_interval::Int)
                 base_logger = TensorBoardLogger.TBLogger(
                     log_dir,
                     (overwrite ? (TensorBoardLogger.tb_overwrite,) : ())...
                 )
                 return SRLogger(; logger=base_logger, log_interval)
             end
-        """
-        )
+        """)
         log_dir = str(self.log_dir)
         return make_logger(log_dir, self.overwrite, self.log_interval)
 

pysr/test/test_cli.py

@@ -15,8 +15,7 @@ def setUp(self):
             self.cli_runner = click_testing.CliRunner()
 
         def test_help_on_all_commands(self):
-            expected = dedent(
-                """
+            expected = dedent("""
                     Usage: pysr [OPTIONS] COMMAND [ARGS]...
 
                     Options:
@@ -25,15 +24,13 @@ def test_help_on_all_commands(self):
                     Commands:
                       install  DEPRECATED (dependencies are now installed at import).
                       test     Run parts of the PySR test suite.
-                """
-            )
+                """)
             result = self.cli_runner.invoke(pysr, ["--help"])
             self.assertEqual(result.output.strip(), expected.strip())
             self.assertEqual(result.exit_code, 0)
 
         def test_help_on_install(self):
-            expected = dedent(
-                """
+            expected = dedent("""
                 Usage: pysr install [OPTIONS]
 
                   DEPRECATED (dependencies are now installed at import).
@@ -44,15 +41,13 @@ def test_help_on_install(self):
                   --precompile
                   --no-precompile
                   --help              Show this message and exit.
-                """
-            )
+                """)
             result = self.cli_runner.invoke(pysr, ["install", "--help"])
             self.assertEqual(result.output.strip(), expected.strip())
             self.assertEqual(result.exit_code, 0)
 
         def test_help_on_test(self):
-            expected = dedent(
-                """
+            expected = dedent("""
                 Usage: pysr test [OPTIONS] TESTS
 
                   Run parts of the PySR test suite.
@@ -63,8 +58,7 @@ def test_help_on_test(self):
                 Options:
                   -k TEXT  Filter expressions to select specific tests.
                   --help   Show this message and exit.
-                """
-            )
+                """)
             result = self.cli_runner.invoke(pysr, ["test", "--help"])
             self.assertEqual(result.output.strip(), expected.strip())
             self.assertEqual(result.exit_code, 0)

pysr/test/test_main.py

@@ -130,16 +130,14 @@ def _multiprocessing_turbo_custom_objective(self, loss_key):
             parallelism="multiprocessing",
             turbo=True,
             early_stop_condition="stop_if(loss, complexity) = loss < 1e-10 && complexity == 1",
-            **{
-                loss_key: f"""
+            **{loss_key: f"""
             function my_objective(tree::{node_type}{{T}}, dataset::Dataset{{T}}, options::Options) where T
                 prediction, flag = eval_tree_array(tree, dataset.X, options)
                 !flag && return T(Inf)
                 abs3(x) = abs(x) ^ 3
                 return sum(abs3, prediction .- dataset.y) / length(prediction)
             end
-            """
-            },
+            """},
         )
         model.fit(self.X, y)
         print(model.equations_)
@@ -154,14 +152,12 @@ def _multiprocessing_turbo_custom_objective(self, loss_key):
 
     def test_multiline_seval(self):
         # The user should be able to run multiple things in a single seval call:
-        num = jl.seval(
-            """
+        num = jl.seval("""
             function my_new_objective(x)
                 x^2
             end
             1.5
-        """
-        )
+        """)
         self.assertEqual(num, 1.5)
 
     def test_high_precision_search_custom_loss(self):

pysr/test/test_startup.py

@@ -69,8 +69,7 @@ def test_warm_start_from_file(self):
                 [
                     sys.executable,
                     "-c",
-                    textwrap.dedent(
-                        f"""
+                    textwrap.dedent(f"""
                         from pysr import PySRRegressor
                         import numpy as np
 
@@ -98,8 +97,7 @@ def test_warm_start_from_file(self):
                         best_loss = model.equations_.iloc[-1]["loss"]
 
                         assert best_loss <= {best_loss}
-                    """
-                    ),
+                    """),
                 ],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,

setup.py

@@ -1,15 +1,13 @@
 # setup.py – retained only for users who still type python setup.py ..."
 import sys
 
-sys.stderr.write(
-    """⚠️  PySR uses pyproject.toml instead of setup.py.
+sys.stderr.write("""⚠️  PySR uses pyproject.toml instead of setup.py.
 
 Install from a checkout with:
     python -m pip install .       # normal
     python -m pip install -e .    # editable (pip ≥21.3)
 
 Or install from PyPI with:
     pip install pysr
-"""
-)
+""")
 sys.exit(1)