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

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

Author: pre-commit-ci[bot]

pysr/sr.py

@@ -1471,8 +1471,10 @@ def __getstate__(self) -> dict[str, Any]:
     def __setstate__(self, state: dict[str, Any]) -> None:
         # ponytail: raise immediately on reload instead of confusing SymPy error later
         self.__dict__.update(state)
-        if "equations_" in state and state["equations_"] is not None and isinstance(
-            self.expression_spec, TemplateExpressionSpec
+        if (
+            "equations_" in state
+            and state["equations_"] is not None
+            and isinstance(self.expression_spec, TemplateExpressionSpec)
         ):
             raise NotImplementedError(
                 "Reloading fitted TemplateExpressionSpec models is not yet supported. "

pysr/test/test_main.py

@@ -2249,6 +2249,7 @@ class TestTemplateExpressionSpec(unittest.TestCase):
     def test_reload_raises_clear_error(self):
         # ponytail: one check — reload of fitted template spec raises immediately
         import pickle
+
         model = PySRRegressor(
             expression_spec=TemplateExpressionSpec(
                 combine="f(x)", expressions=["f"], variable_names=["x"]