edge case writer??

rlberry-py · Feb 19, 2024 · 3633bb5 · 3633bb5
1 parent 5569986
commit 3633bb5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rlberry/utils/writers.py b/rlberry/utils/writers.py
@@ -112,6 +112,9 @@ def data(self):
                     [df, pd.DataFrame(self._data[tag])],
                     ignore_index=True,
                 )
+        if df is None:
+            # if there are no data, return empty dataframe
+            df = pd.DataFrame()
         return df
 
     def set_max_global_step(self, max_global_step):