Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 10, 2024
1 parent 4a1f17c commit 1ef2fc6
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pysr/test/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -750,7 +750,7 @@ def test_sequence_short_data_error(self):
)
with self.assertRaises(ValueError):
model.fit(X)

def test_sequence_bad_weight_length_error(self):
X = np.zeros((10, 1))
model = PySRSequenceRegressor(
Expand All @@ -761,7 +761,7 @@ def test_sequence_bad_weight_length_error(self):

def test_sequence_weights(self):
X = np.ones((100, 1))
weights = np.ones((100, ))
weights = np.ones((100,))
model = PySRSequenceRegressor(
recursive_history_length=2,
early_stop_condition="stop_if(loss, complexity) = loss < 1e-4 && complexity == 1",
Expand Down Expand Up @@ -810,7 +810,7 @@ def test_sequence_from_file(self):
n_features_in=2,
recursive_history_length=2,
feature_names_in=["xt_1", "xt_2"],
selection_mask=np.ones(2, dtype=np.bool_)
selection_mask=np.ones(2, dtype=np.bool_),
)
self.assertIn("xt_1", model4.get_best()["equation"])

Expand Down

0 comments on commit 1ef2fc6

Please sign in to comment.