update trainer fixtures

janelia-cellmap · Feb 18, 2025 · 6e613e5 · 6e613e5
1 parent 3f5cd5b
commit 6e613e5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/fixtures/trainers.py b/tests/fixtures/trainers.py
@@ -6,15 +6,12 @@
 @pytest.fixture()
 def dummy_trainer():
     yield DummyTrainerConfig(
-        name="dummy_trainer", learning_rate=1e-5, batch_size=10, mirror_augment=True
+        name="dummy_trainer", dummy_attr=True
     )
 
 
 @pytest.fixture()
 def gunpowder_trainer():
     yield GunpowderTrainerConfig(
         name="default_gp_trainer",
-        batch_size=1,
-        learning_rate=0.0001,
-        num_data_fetchers=1,
     )