fix(metrological_streams): fixed computational error for batches of s…

…ize more than one
Met4FoF · Feb 26, 2021 · 686bad5 · 686bad5
1 parent d0c5ecb
commit 686bad5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agentMET4FOF/metrological_streams.py b/agentMET4FOF/metrological_streams.py
@@ -154,7 +154,10 @@ def _next_sample_generator(self, batch_size: int = 1) -> np.ndarray:
         time uncertainty ``ut`` and measurement uncertainty ``uv`` to sample
         """
         _time: np.ndarray = (
-            np.arange(self._sample_idx, self._sample_idx + batch_size, 1.0).reshape(-1, 1) / self.sfreq
+            np.arange(self._sample_idx, self._sample_idx + batch_size, 1.0).reshape(
+                -1, 1
+            )
+            / self.sfreq
         )
         self._sample_idx += batch_size