Spaces:
Running
Running
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")
|