MilesCranmer commited on
Commit
86421d6
1 Parent(s): 041087e

Avoid complex data check in sklearn test

Browse files
Files changed (1) hide show
  1. 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")