Merge pull request #131 from MilesCranmer/deepsource-transform-9e2a0e1f

benchmarks/space.py

@@ -62,7 +62,9 @@ space = dict(
         "crossover_probability", np.log(0.00001), np.log(0.2)
     ),
     #     perturbation_factor=1.0,
-    perturbation_factor=hp.loguniform("perturbation_factor", np.log(0.0001), np.log(100)),
+    perturbation_factor=hp.loguniform(
+        "perturbation_factor", np.log(0.0001), np.log(100)
+    ),
     #     maxsize=20,
     maxsize=hp.choice("maxsize", [30]),
     #     warmup_maxsize_by=0.0,

pysr/deprecated.py

@@ -1,5 +1,6 @@
 """Various functions to deprecate features."""
 
+
 def make_deprecated_kwargs_for_pysr_regressor():
     """Create dict of deprecated kwargs."""
 
@@ -33,4 +34,4 @@ def make_deprecated_kwargs_for_pysr_regressor():
         old, new = line.split("=>")
         deprecated_kwargs[old] = new
 
-    return deprecated_kwargs
+    return deprecated_kwargs

test/test.py

@@ -268,6 +268,7 @@ class TestFeatureSelection(unittest.TestCase):
 
 class TestMiscellaneous(unittest.TestCase):
     """Test miscellaneous functions."""
+
     def test_deprecation(self):
         # Ensure that deprecation works as expected, with a warning,
         # and sets the correct value.
@@ -276,4 +277,4 @@ class TestMiscellaneous(unittest.TestCase):
         # This is a deprecated parameter, so we should get a warning.
 
         # The correct value should be set:
-        self.assertEqual(model.params["fraction_replaced"], 0.2)
+        self.assertEqual(model.params["fraction_replaced"], 0.2)