Fix test in training
@@ -281,8 +281,9 @@ def test_model_methods():
out = model.predict([input_a_np, input_b_np], batch_size=4)
# empty batch
- with pytest.raises(StopIteration):
+ with pytest.raises(ValueError):
def gen_data():
+ while True:
yield (np.asarray([]), np.asarray([]))
out = model.evaluate_generator(gen_data(), steps=1)
