tttc3 commited on
Commit
f570132
1 Parent(s): 4819728

Ensure iteration completes for multioutput test

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