fix: update logging messages for clarity and consistency in ForwardPass and ForwardPassStrategy

Author: bnb32

sup3r/pipeline/forward_pass.py

@@ -448,7 +448,7 @@ def run(cls, strategy, node_index):
         """
         if not strategy.node_finished(node_index):
             logger.info(
-                'Starting forward pass on node %s with %s chunks using %s '
+                'Starting forward pass on node %s with %s chunk(s) using %s '
                 'execution.',
                 node_index,
                 len(strategy.node_chunks[node_index]),
@@ -517,7 +517,7 @@ def _run_serial(cls, strategy, node_index):
                     raise MemoryError(msg)
 
         logger.info(
-            'Finished forward passes on %s chunks in %s',
+            'Finished forward pass(es) on %s chunk(s) in %s',
             len(strategy.node_chunks[node_index]),
             dt.now() - start,
         )

sup3r/pipeline/strategy.py

@@ -468,7 +468,7 @@ def preflight(self):
         non_masked = self.fwp_slicer.n_spatial_chunks - sum(self.fwp_mask)
         non_masked *= self.fwp_slicer.n_time_chunks
         logger.info(
-            'Chunk strategy uses %s nodes across %s total chunks '
+            'Chunk strategy uses %s node(s) across %s chunk(s): '
             '(%s spatial x %s temporal, %s unmasked).',
             len(self.node_chunks),
             self.fwp_slicer.n_chunks,

sup3r/preprocessing/samplers/base.py

@@ -191,12 +191,15 @@ def preflight(self):
             'building batches with n_samples = batch_size, each with '
             'n_time_steps = sample_shape[2].'
         )
-        if self.data.shape[2] < self.sample_shape[2] * self.batch_size:
+        if (
+            self.data.shape[2] < self.sample_shape[2] * self.batch_size
+            and self.data.shape[2] > 1
+        ):
             logger.warning(msg)
             warn(msg)
 
         if self.mode == 'eager':
-            logger.info('Received mode = "eager".')
+            logger.debug('Received mode = "eager".')
             _ = self.compute()
 
     def check_proxy_obs_consistency(self):