From c90ac9c3fbc318f29dbedb793147b354e007ba21 Mon Sep 17 00:00:00 2001 From: Okunator Date: Wed, 4 Jan 2023 18:24:31 +0200 Subject: [PATCH] test(training): fix failing test --- cellseg_models_pytorch/training/tests/test_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cellseg_models_pytorch/training/tests/test_training.py b/cellseg_models_pytorch/training/tests/test_training.py index f835858..5a4f443 100644 --- a/cellseg_models_pytorch/training/tests/test_training.py +++ b/cellseg_models_pytorch/training/tests/test_training.py @@ -9,7 +9,7 @@ from cellseg_models_pytorch.training.lit import SegmentationExperiment -def test_training(img_patch_dir, mask_patch_dir, auto_lr): +def test_training(img_patch_dir, mask_patch_dir): train_ds = SegmentationFolderDataset( path=img_patch_dir.as_posix(), mask_path=mask_patch_dir.as_posix(),