Fix TO_DATE feature

defog-ai · Aug 5, 2024 · ae36c54 · ae36c54
1 parent ffcd651
commit ae36c54
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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):