[ref] It is now possible to record the time label from a stream using…

… the Hdf_recorder Block
LaboratoireMecaniqueLille · Mar 7, 2023 · 4ccdd67 · 4ccdd67
1 parent f956199
commit 4ccdd67
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crappy/inout/t7Streamer.py b/crappy/inout/t7Streamer.py
@@ -263,7 +263,7 @@ def get_stream(self) -> Optional[List[np.ndarray]]:
                                     data.shape[0]) / self._scan_rate
     self._n_points += data.shape[0]
 
-    return [t, data]
+    return [t[:, np.newaxis], data]
 
   def stop_stream(self) -> None:
     """Stops the stream if it was started."""

diff --git a/crappy/modifier/demux.py b/crappy/modifier/demux.py
@@ -80,6 +80,6 @@ def evaluate(self, data: Dict[str, np.ndarray]) -> Dict[str, Any]:
     if self._mean:
       data[self._time_label] = np.mean(data[self._time_label])
     else:
-      data[self._time_label] = data[self._time_label][0]
+      data[self._time_label] = np.squeeze(data[self._time_label])[0]
 
     return data