Fix unexpected additional x scaler arg

FR-DC · Jun 10, 2024 · 8497c3e · 8497c3e
1 parent 02a4d57
commit 8497c3e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/integration_tests/test_pipeline.py b/tests/integration_tests/test_pipeline.py
@@ -48,7 +48,6 @@ def test_manual_segmentation_pipeline(model_fn, ds):
         in_channels=ds.ar.shape[-1],
         n_classes=n_classes,
         lr=1e-3,
-        x_scaler=ss,
         y_encoder=oe,
         frozen=True,
     )