test(training): fix failing test

okunator · Jan 4, 2023 · c90ac9c · c90ac9c
1 parent d8ddad4
commit c90ac9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(),