Fix issue with parsing pandas.Timestamp

e2b-dev · Jan 9, 2025 · 6642231 · 6642231
1 parent 8ced45b
commit 6642231
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/.changeset/itchy-ducks-breathe.md b/.changeset/itchy-ducks-breathe.md
@@ -0,0 +1,5 @@
+---
+'@e2b/code-interpreter-template': patch
+---
+
+Fix timestamp issue
diff --git a/template/startup_scripts/0002_data.py b/template/startup_scripts/0002_data.py
@@ -20,7 +20,13 @@ def _figure_repr_e2b_chart_(self: Figure):
 
 
 def _dataframe_repr_e2b_data_(self: pandas.DataFrame):
-    return orjson.loads(orjson.dumps(self.to_dict(orient="list")))
+    result = self.to_dict(orient="list")
+    for key, value in result.items():
+        # Check each column's values
+        result[key] = [
+            v.isoformat() if isinstance(v, pandas.Timestamp) else v for v in value
+        ]
+    return result
 
 
 class E2BDataFormatter(BaseFormatter):