Update __init__.py

AutoResearch · Jun 21, 2024 · f80ee53 · f80ee53
1 parent f74d07f
commit f80ee53
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/autora/experiment_runner/experimentation_manager/firebase/__init__.py b/src/autora/experiment_runner/experimentation_manager/firebase/__init__.py
@@ -32,6 +32,8 @@ def _sequence_to_db_object(iterable, save=False):
     """
     if not hasattr(iterable, "__iter__"):
         return {0: iterable}
+    if isinstance(iterable, pd.DataFrame):
+        return iterable.to_dict(orient='index')
     is_int64 = False
     is_float64 = False
     for t in iterable: