proper fix for future warning on empty dfs

assume-framework · Sep 25, 2023 · 2aa152a · 2aa152a
1 parent 8b17b3e
commit 2aa152a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/assume/common/outputs.py b/assume/common/outputs.py
@@ -204,7 +204,6 @@ async def store_dfs(self):
                 continue
 
             # exclude all empty dataframes from self.write_dfs[table]
-            self.write_dfs[table] = [df for df in self.write_dfs[table] if not df.empty]
             df = pd.concat(self.write_dfs[table], axis=0)
 
             df.reset_index()
@@ -322,7 +321,8 @@ def write_market_dispatch(self, data):
         """
         df = pd.DataFrame(data, columns=["datetime", "power", "market_id", "unit_id"])
         df["simulation"] = self.simulation_id
-        self.write_dfs["market_dispatch"].append(df)
+        if not df.empty:
+            self.write_dfs["market_dispatch"].append(df)
 
     def write_unit_dispatch(self, data):
         """