Merge pull request #487 from gipert/flow

Bug fix in `FileDB.from_disk()`
legend-exp · May 5, 2023 · a492e41 · a492e41
2 parents 491df43 + 973d1b1
commit a492e41
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/pygama/flow/file_db.py b/src/pygama/flow/file_db.py
@@ -533,9 +533,11 @@ def from_disk(self, path: str | list[str]) -> None:
             # read in dataframe
             df = pd.read_hdf(p, key="dataframe")
 
+            # first iteration
             if _columns is None:
                 _columns = columns
                 _df = df
+                continue
             elif _columns != columns:
                 log.debug("found inconsistent FileDB, trying to merge")
                 # if columns are not the same, need to merge the two dataframes
@@ -561,9 +563,9 @@ def _replace_idx(row, idx, new_idx, tier):
                                 args=(idx, new_idx, tier),
                                 axis=1,
                             )
-            else:
-                # if columns are the same, assume we can safely concat the dataframes
-                _df = pd.concat([_df, df], ignore_index=True, copy=False)
+
+            # now we can safely concat the dataframes
+            _df = pd.concat([_df, df], ignore_index=True, copy=False)
 
         self.set_config(_cfg)
         self.df = _df