From ae36c54e2665bb99e6fb357f30b78c5db0f1bad7 Mon Sep 17 00:00:00 2001
From: jp <wongjingping@gmail.com>
Date: Mon, 5 Aug 2024 10:00:07 +0800
Subject: [PATCH] Fix TO_DATE feature

---
 defog_utils/utils_sql.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/defog_utils/utils_sql.py b/defog_utils/utils_sql.py
index 0bc09bc..47797e5 100644
--- a/defog_utils/utils_sql.py
+++ b/defog_utils/utils_sql.py
@@ -429,6 +429,8 @@ def get_sql_features(
             features.date_sub = True
         elif isinstance(node, exp.DateTrunc) or isinstance(node, exp.TimestampTrunc):
             features.date_trunc = True
+        elif isinstance(node, exp.StrToDate):
+            features.date_time_type_conversion = True
         elif isinstance(node, exp.StrToTime):
             features.date_time_type_conversion = True
         elif isinstance(node, exp.Extract):