Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
86421d6
1
Parent(s):
041087e
Avoid complex data check in sklearn test
Browse files- pysr/test/test.py +3 -0
pysr/test/test.py
CHANGED
@@ -678,6 +678,9 @@ class TestMiscellaneous(unittest.TestCase):
|
|
678 |
check_generator = check_estimator(model, generate_only=True)
|
679 |
exception_messages = []
|
680 |
for _, check in check_generator:
|
|
|
|
|
|
|
681 |
try:
|
682 |
with warnings.catch_warnings():
|
683 |
warnings.simplefilter("ignore")
|
|
|
678 |
check_generator = check_estimator(model, generate_only=True)
|
679 |
exception_messages = []
|
680 |
for _, check in check_generator:
|
681 |
+
if check.func.__name__ == "check_complex_data":
|
682 |
+
# We can use complex data, so avoid this check.
|
683 |
+
continue
|
684 |
try:
|
685 |
with warnings.catch_warnings():
|
686 |
warnings.simplefilter("ignore")
|