fix mypy

Author: jialuoo

bigframes/core/compile/sqlglot/expressions/datetime_ops.py

@@ -71,7 +71,7 @@ def _datetime_to_integer_label_fixed_frequency(
         "UNIX_MICROS",
         sge.Cast(this=x.expr, to=sge.DataType(this=sge.DataType.Type.TIMESTAMPTZ)),
     )
-    first = _calculate_resample_first(y, op.origin)
+    first = _calculate_resample_first(y, op.origin)  # type: ignore
     x_int_label = sge.Cast(
         this=sge.Floor(
             this=sge.func(
@@ -147,7 +147,7 @@ def _datetime_to_integer_label_non_fixed_frequency(
             ),
         )
     elif rule_code == "ME":  # Monthly
-        x_int = sge.Paren(
+        x_int = sge.Paren(  # type: ignore
             this=sge.Add(
                 this=sge.Mul(
                     this=sge.Extract(
@@ -163,7 +163,7 @@ def _datetime_to_integer_label_non_fixed_frequency(
                 ),
             )
         )
-        first = sge.Paren(
+        first = sge.Paren(  # type: ignore
             this=sge.Add(
                 this=sge.Mul(
                     this=sge.Extract(
@@ -204,7 +204,7 @@ def _datetime_to_integer_label_non_fixed_frequency(
             ),
         )
     elif rule_code == "QE-DEC":  # Quarterly
-        x_int = sge.Paren(
+        x_int = sge.Paren(  # type: ignore
             this=sge.Add(
                 this=sge.Mul(
                     this=sge.Extract(
@@ -220,7 +220,7 @@ def _datetime_to_integer_label_non_fixed_frequency(
                 ),
             )
         )
-        first = sge.Paren(
+        first = sge.Paren(  # type: ignore
             this=sge.Add(
                 this=sge.Mul(
                     this=sge.Extract(

tests/unit/core/compile/sqlglot/expressions/test_datetime_ops.py

@@ -62,21 +62,21 @@ def test_datetime_to_integer_label(scalar_types_df: bpd.DataFrame, snapshot):
     bf_df = scalar_types_df[col_names]
     ops_map = {
         "fixed_freq": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.Day(), origin="start", closed="left"
+            freq=pd.tseries.offsets.Day(), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
         "non_fixed_freq_monthly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.MonthEnd(), origin="start", closed="left"
+            freq=pd.tseries.offsets.MonthEnd(), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
         "non_fixed_freq_weekly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.Week(weekday=6), origin="start", closed="left"
+            freq=pd.tseries.offsets.Week(weekday=6), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
         "non_fixed_freq_quarterly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.QuarterEnd(startingMonth=12),
+            freq=pd.tseries.offsets.QuarterEnd(startingMonth=12),  # type: ignore
             origin="start",
             closed="left",
         ).as_expr("datetime_col", "timestamp_col"),
         "non_fixed_freq_yearly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.YearEnd(month=12), origin="start", closed="left"
+            freq=pd.tseries.offsets.YearEnd(month=12), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
     }