fix test coverage

quintenroets · Apr 14, 2024 · 9d0ad14 · 9d0ad14
1 parent 5285fce
commit 9d0ad14
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/src/revnets/models/config.py b/src/revnets/models/config.py
@@ -70,14 +70,14 @@ class Config(SerializationMixin):
     console_metrics_refresh_interval: float = 0.5
     max_difficult_inputs_epochs: int = 100
 
+    limit_batches: int | None = None
+
     @property
     def number_of_validation_sanity_steps(self) -> int | None:
         return 0 if self.debug else 0
 
-    @property
-    def limit_batches(self) -> int | None:
-        return self.debug_batch_limit if self.debug else None
-
     def __post_init__(self) -> None:
         if self.evaluation.run_analysis:
             self.always_train = False
+        if self.debug:
+            self.limit_batches = self.debug_batch_limit  # pragma: nocover
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -34,6 +34,7 @@ def test_context(context: Context, mocked_assets_path: None) -> Iterator[Context
         reconstruction_training=hyperparameters,
         max_difficult_inputs_epochs=1,
         evaluation=evaluation,
+        limit_batches=5,
     )
     mock = patch("matplotlib.pyplot.show")
     with mock:

diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -15,10 +15,11 @@
     *pipeline_modules,
     pipelines.mininet.mininet_40,
     pipelines.mininet.mininet_100,
+    pipelines.mediumnet.mediumnet_40,
     pipelines.images.mininet_100,
     pipelines.images.mininet_128,
     pipelines.images.mininet_200,
-    pipelines.mediumnet.mediumnet_40,
+    pipelines.images.mediumnet,
 )