allow int timedelta to micro

Author: chelsea-lin

bigframes/core/compile/sqlglot/sqlglot_ir.py

@@ -633,8 +633,6 @@ def _literal(value: typing.Any, dtype: dtypes.Dtype) -> sge.Expression:
     elif dtype == dtypes.JSON_DTYPE:
         return sge.ParseJSON(this=sge.convert(str(value)))
     elif dtype == dtypes.TIMEDELTA_DTYPE:
-        if isinstance(value, int):
-            return sge.convert(value)
         return sge.convert(utils.timedelta_to_micros(value))
     elif dtypes.is_struct_like(dtype):
         items = [

bigframes/core/utils.py

@@ -234,7 +234,7 @@ def wrapper(*args, **kwargs):
 
 
 def timedelta_to_micros(
-    timedelta: typing.Union[pd.Timedelta, datetime.timedelta, np.timedelta64]
+    timedelta: typing.Union[pd.Timedelta, datetime.timedelta, np.timedelta64, int]
 ) -> int:
     if isinstance(timedelta, pd.Timedelta):
         # pd.Timedelta.value returns total nanoseconds.
@@ -248,4 +248,7 @@ def timedelta_to_micros(
             (timedelta.days * 3600 * 24) + timedelta.seconds
         ) * 1_000_000 + timedelta.microseconds
 
+    if isinstance(timedelta, int):
+        return timedelta
+
     raise TypeError(f"Unrecognized input type: {type(timedelta)}")