fix typping

tsugumi-sys · Jan 15, 2024 · 7be8106 · 7be8106
1 parent 0d8ff33
commit 7be8106
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/pipelines/test_experimenter.py b/tests/pipelines/test_experimenter.py
@@ -16,7 +16,6 @@ def test_run():
         model = TestModel()
         training_params: TrainingParams = {
             "epochs": 1,
-            "batch_size": 1,
             "loss_criterion": nn.MSELoss(),
             "accuracy_criterion": nn.L1Loss(),
             "optimizer": Adam(model.parameters(), lr=0.0005),