From f18d16b75905c175e1b5820290c044493fe27e52 Mon Sep 17 00:00:00 2001
From: Ayush Dattagupta <ayushdg95@gmail.com>
Date: Tue, 26 Apr 2022 16:21:08 -0700
Subject: [PATCH] Update reinterpret and timedelta to explicitly cast to int64
 instead of int

---
 dask_sql/mappings.py               | 6 +-----
 dask_sql/physical/rex/core/call.py | 2 +-
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/dask_sql/mappings.py b/dask_sql/mappings.py
index 5890df279..5b35e391b 100644
--- a/dask_sql/mappings.py
+++ b/dask_sql/mappings.py
@@ -196,10 +196,6 @@ def sql_to_python_type(sql_type: str) -> type:
     if sql_type.startswith("CHAR(") or sql_type.startswith("VARCHAR("):
         return pd.StringDtype()
     elif sql_type.startswith("INTERVAL"):
-        interval_type = sql_type.split()[1].lower()
-        if interval_type in {"year", "quarter", "month"}:
-            # if sql_type is INTERVAL YEAR, Calcite will covert to months
-            return np.dtype("timedelta64[M]")
         return np.dtype("<m8[ns]")
 
     elif sql_type.startswith("TIMESTAMP(") or sql_type.startswith("TIME("):
@@ -304,7 +300,7 @@ def cast_column_to_type(col: dd.Series, expected_type: str):
         col = da.trunc(col.fillna(value=np.NaN))
 
     if current_timedelta_type and expected_integer:
-        res = col.astype(int)
+        res = col.astype(np.int64)
     else:
         res = col.astype(expected_type)
 
diff --git a/dask_sql/physical/rex/core/call.py b/dask_sql/physical/rex/core/call.py
index 322ce015c..a3b8faf67 100644
--- a/dask_sql/physical/rex/core/call.py
+++ b/dask_sql/physical/rex/core/call.py
@@ -173,7 +173,7 @@ def div(self, lhs, rhs):
             return result
         else:  # pragma: no cover
             result = da.trunc(result)
-            result = result.astype(int)
+            result = result.astype(np.int64)
             return result