diff --git a/tests/rs_models/test_seg_models.py b/tests/rs_models/test_seg_models.py index af948d4..4b19f83 100644 --- a/tests/rs_models/test_seg_models.py +++ b/tests/rs_models/test_seg_models.py @@ -50,7 +50,7 @@ class TestFarSegModel(TestSegModel): def set_specs(self): self.specs = [ - dict(), dict(num_classes=20), dict(encoder_pretrained=False), + dict(), dict(num_classes=20), dict(pretrained_encoder=False), dict(in_channels=10) ]