Spaces:
Sleeping
Sleeping
tttc3
commited on
Commit
•
f570132
1
Parent(s):
4819728
Ensure iteration completes for multioutput test
Browse files- test/test.py +6 -0
test/test.py
CHANGED
@@ -366,6 +366,12 @@ class TestMiscellaneous(unittest.TestCase):
|
|
366 |
try:
|
367 |
with warnings.catch_warnings():
|
368 |
warnings.simplefilter("ignore")
|
|
|
|
|
|
|
|
|
|
|
|
|
369 |
check(model)
|
370 |
print("Passed", check.func.__name__)
|
371 |
except Exception as e:
|
|
|
366 |
try:
|
367 |
with warnings.catch_warnings():
|
368 |
warnings.simplefilter("ignore")
|
369 |
+
# To ensure an equation file is written for each output in
|
370 |
+
# nout, set stop condition to niterations=1
|
371 |
+
if check.func.__name__ == "check_regressor_multioutput":
|
372 |
+
model.set_params(niterations=1, max_evals=None)
|
373 |
+
else:
|
374 |
+
model.set_params(max_evals=10000)
|
375 |
check(model)
|
376 |
print("Passed", check.func.__name__)
|
377 |
except Exception as e:
|