Skip to content

Commit b3dec0f

Browse files
committed
update kwarg test to run with pb display
Signed-off-by: GiulioZizzo <[email protected]>
1 parent 7bed09d commit b3dec0f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/estimators/classification/test_deeplearning_common.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def test_functional_model(art_warning, image_dl_estimator):
191191
art_warning(e)
192192

193193

194-
@pytest.mark.skip_framework("mxnet", "tensorflow", "pytorch", "huggingface", "non_dl_frameworks")
194+
@pytest.mark.skip_framework("mxnet", "non_dl_frameworks")
195195
def test_fit_kwargs(art_warning, image_dl_estimator, get_default_mnist_subset, default_batch_size):
196196
try:
197197
(x_train_mnist, y_train_mnist), (_, _) = get_default_mnist_subset
@@ -201,7 +201,8 @@ def get_lr(_):
201201

202202
# Test a valid callback
203203
classifier, _ = image_dl_estimator(from_logits=True)
204-
kwargs = {"callbacks": [LearningRateScheduler(get_lr)]}
204+
kwargs = {"callbacks": [LearningRateScheduler(get_lr)],
205+
"display_progress_bar": True}
205206
classifier.fit(x_train_mnist, y_train_mnist, batch_size=default_batch_size, nb_epochs=1, **kwargs)
206207

207208
# Test failure for invalid parameters

0 commit comments

Comments
 (0)