Fix test

epochlib/data/pipeline_dataset.py

@@ -77,7 +77,7 @@ def setup_pipeline(self, *, use_augmentations: bool) -> None:
 
         :param use_augmentations: Whether to use augmentations while passing data through pipeline
         """
-        self._enabled_steps = []
+        self._enabled_steps: list[TrainingBlock] = []
 
         if self.steps is not None:
             for step in self.steps: