Fix regression with timestamp field with pyarrow backend

polyaxon · Jun 30, 2023 · 9469531 · 9469531
1 parent f5d3918
commit 9469531
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/traceml/traceml/events/schemas.py b/traceml/traceml/events/schemas.py
@@ -388,6 +388,9 @@ def read(
                     sep=V1Event._SEPARATOR,
                     engine="pyarrow",
                 )
+                # Pyarrow automatically converts timestamp fields
+                if "timestamp" in df.columns:
+                    df["timestamp"] = df["timestamp"].astype(str)
         elif isinstance(data, dict):
             df = pd.DataFrame.from_dict(data)
         else: