diff --git a/traceml/traceml/events/schemas.py b/traceml/traceml/events/schemas.py index 0929dc930..36f47a101 100644 --- 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: