Merge pull request #212 from ImperialCollegeLondon/fix-biologic-mb-im…

…port-errors Fix biologic mb import errors
ImperialCollegeLondon · Jan 22, 2025 · a97ebb7 · a97ebb7
2 parents 7308a58 + dbd16c9
commit a97ebb7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyprobe/cyclers/basecycler.py b/pyprobe/cyclers/basecycler.py
@@ -210,7 +210,7 @@ def _get_dataframe_list(self) -> list[pl.DataFrame | pl.LazyFrame]:
         all_columns = set([col for df in list for col in df.collect_schema().names()])
         for i in range(len(list)):
             if len(list[i].collect_schema().names()) < len(all_columns):
-                warnings.warn(
+                logger.warning(
                     f"File {os.path.basename(files[i])} has missing columns, "
                     "these have been filled with null values."
                 )

diff --git a/pyprobe/cyclers/biologic.py b/pyprobe/cyclers/biologic.py
@@ -114,7 +114,7 @@ def get_imported_dataframe(
         for i, df in enumerate(dataframe_list):
             df = df.with_columns(pl.lit(i).alias("MB File"))
             df_list.append(df)
-        complete_df = pl.concat(df_list, how="vertical")
+        complete_df = pl.concat(df_list, how="diagonal")
         complete_df = self.apply_step_correction(complete_df)
         return complete_df