From b5448a4e16a85f10324c2177d2bd7a63acc9a248 Mon Sep 17 00:00:00 2001
From: Wesley Gifford <79663411+wgifford@users.noreply.github.com>
Date: Tue, 12 Nov 2024 15:37:56 -0500
Subject: [PATCH] do not enforce datetype

---
 tsfm_public/toolkit/dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tsfm_public/toolkit/dataset.py b/tsfm_public/toolkit/dataset.py
index d041dd16..3849564c 100644
--- a/tsfm_public/toolkit/dataset.py
+++ b/tsfm_public/toolkit/dataset.py
@@ -88,8 +88,8 @@ def __init__(
 
         # sort the data by datetime
         if timestamp_column in list(data_df.columns):
-            if not isinstance(data_df[timestamp_column].iloc[0], pd.Timestamp):
-                data_df[timestamp_column] = pd.to_datetime(data_df[timestamp_column])
+            # if not isinstance(data_df[timestamp_column].iloc[0], pd.Timestamp):
+            #     data_df[timestamp_column] = pd.to_datetime(data_df[timestamp_column])
             data_df = data_df.sort_values(timestamp_column, ignore_index=True)
 
         # pad zero to the data_df if the len is shorter than seq_len+pred_len