From 2ca66406e4765d9ca0f3686c79463941d2c9214e Mon Sep 17 00:00:00 2001 From: bw4sz Date: Fri, 1 Dec 2023 19:35:22 -0800 Subject: [PATCH] we no longer need two sets of tests for image_callback --- tests/test_callbacks.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/tests/test_callbacks.py b/tests/test_callbacks.py index 85ec2291..639d5557 100644 --- a/tests/test_callbacks.py +++ b/tests/test_callbacks.py @@ -14,14 +14,6 @@ def test_log_images(m, every_n_epochs, tmpdir): m.trainer.fit(m) saved_images = glob.glob("{}/*.png".format(tmpdir)) assert len(saved_images) == 1 - -@pytest.mark.parametrize("every_n_epochs", [1, 2, 3]) -def test_log_images_multiclass(two_class_m, every_n_epochs, tmpdir): - im_callback = callbacks.images_callback(savedir=tmpdir, every_n_epochs=every_n_epochs) - two_class_m.create_trainer(callbacks=[im_callback]) - two_class_m.trainer.fit(two_class_m) - saved_images = glob.glob("{}/*.png".format(tmpdir)) - assert len(saved_images) == 1 def test_create_checkpoint(m, tmpdir): checkpoint_callback = ModelCheckpoint(