Skip to content

Commit 29e4846

Browse files
ensure valid interval return
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 533f74b commit 29e4846

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

src/databricks/sql/backend/sea/utils/normalize.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,10 @@ def normalize_sea_type_to_thrift(type_name: str, col_data: Dict[str, Any]) -> st
4141
if type_name == "INTERVAL":
4242
type_interval_type = col_data.get("type_interval_type")
4343
if type_interval_type:
44-
if any(t in type_interval_type.upper() for t in ["YEAR", "MONTH"]):
45-
return "INTERVAL_YEAR_MONTH"
46-
elif any(
47-
t in type_interval_type.upper()
48-
for t in ["DAY", "HOUR", "MINUTE", "SECOND"]
49-
):
50-
return "INTERVAL_DAY_TIME"
44+
return (
45+
"INTERVAL_YEAR_MONTH"
46+
if any(t in type_interval_type.upper() for t in ["YEAR", "MONTH"])
47+
else "INTERVAL_DAY_TIME"
48+
)
5149

5250
return normalized_type

0 commit comments

Comments
 (0)