Skip to content

Commit

Permalink
test: skip new sklearn checks
Browse files Browse the repository at this point in the history
  • Loading branch information
MilesCranmer committed Dec 12, 2024
1 parent 89b5a89 commit 85b4f40
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions pysr/test/test_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -876,8 +876,14 @@ def test_scikit_learn_compatibility(self):
check_generator = check_estimator(model, generate_only=True)
exception_messages = []
for _, check in check_generator:
if check.func.__name__ == "check_complex_data":
# We can use complex data, so avoid this check.
if check.func.__name__ in {
# We can use complex data, so avoid this check
"check_complex_data",
# We handle kwargs manually, so skip this check
"check_do_not_raise_errors_in_init_or_set_params",
# TODO:
"check_n_features_in_after_fitting",
}:
continue
try:
with warnings.catch_warnings():
Expand Down

0 comments on commit 85b4f40

Please sign in to comment.