fix save when no batches were processed

py_neuromodulation/stream/stream.py

@@ -38,7 +38,6 @@ def __init__(
         sampling_rate_features_hz: float | None = None,
         path_grids: _PathLike | None = None,
         coord_names: list | None = None,
-        is_stream_lsl: bool = False,
         coord_list: list | None = None,
         verbose: bool = True,
     ) -> None:

py_neuromodulation/utils/file_writer.py

@@ -4,6 +4,7 @@
 import pandas as pd
 import numpy as np
 from py_neuromodulation.utils.types import _PathLike
+from py_neuromodulation import logger
 
 class AbstractFileWriter(ABC):
 
@@ -78,7 +79,8 @@ def load_all(self):
         for i in range(self.idx):
             with open(self.out_dir / f"{self.name}-{i}.msgpack", "rb") as f:
                 data_l.append(msgpack.unpack(f))
-
+        if len(self.data_l) == 0:
+            raise ValueError("No data to load")
         data = pd.DataFrame(list(np.concatenate(data_l)))
         return data
 
@@ -88,7 +90,11 @@ def save_as_csv(self, save_all_combined: bool = False):
         """
 
         if save_all_combined:
-            data = self.load_all()
+            try:
+                data = self.load_all()
+            except ValueError as e:
+                logger.error(e)
+                return
             data.to_csv(self.csv_path, index=False)
         else:
             if len(self.data_l) > 0: