Skip to content

Commit 638f083

Browse files
committed
chore: adjusts type hints to account for complications with mypy part 6
1 parent cc20328 commit 638f083

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

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

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -62,37 +62,37 @@ def test_datetime_to_integer_label(scalar_types_df: bpd.DataFrame, snapshot):
6262
bf_df = scalar_types_df[col_names]
6363
ops_map = {
6464
"fixed_freq": ops.DatetimeToIntegerLabelOp(
65-
freq=pd.tseries.offsets.Day(),
65+
freq=pd.tseries.offsets.Day(), # type: ignore[arg-type]
6666
origin="start",
6767
closed="left", # type: ignore
6868
).as_expr("datetime_col", "timestamp_col"),
6969
"origin_epoch": ops.DatetimeToIntegerLabelOp(
70-
freq=pd.tseries.offsets.Day(),
70+
freq=pd.tseries.offsets.Day(), # type: ignore[arg-type]
7171
origin="epoch",
7272
closed="left", # type: ignore
7373
).as_expr("datetime_col", "timestamp_col"),
7474
"origin_start_day": ops.DatetimeToIntegerLabelOp(
75-
freq=pd.tseries.offsets.Day(),
75+
freq=pd.tseries.offsets.Day(), # type: ignore[arg-type]
7676
origin="start_day",
7777
closed="left", # type: ignore
7878
).as_expr("datetime_col", "timestamp_col"),
7979
"non_fixed_freq_weekly": ops.DatetimeToIntegerLabelOp(
80-
freq=pd.tseries.offsets.Week(weekday=6),
80+
freq=pd.tseries.offsets.Week(weekday=6), # type: ignore[arg-type]
8181
origin="start",
8282
closed="left", # type: ignore
8383
).as_expr("datetime_col", "timestamp_col"),
8484
"non_fixed_freq_monthly": ops.DatetimeToIntegerLabelOp(
85-
freq=pd.tseries.offsets.MonthEnd(),
85+
freq=pd.tseries.offsets.MonthEnd(), # type: ignore[arg-type]
8686
origin="start",
8787
closed="left", # type: ignore
8888
).as_expr("datetime_col", "timestamp_col"),
8989
"non_fixed_freq_quarterly": ops.DatetimeToIntegerLabelOp(
90-
freq=pd.tseries.offsets.QuarterEnd(startingMonth=12),
90+
freq=pd.tseries.offsets.QuarterEnd(startingMonth=12), # type: ignore[arg-type]
9191
origin="start",
9292
closed="left", # type: ignore
9393
).as_expr("datetime_col", "timestamp_col"),
9494
"non_fixed_freq_yearly": ops.DatetimeToIntegerLabelOp(
95-
freq=pd.tseries.offsets.YearEnd(),
95+
freq=pd.tseries.offsets.YearEnd(), # type: ignore[arg-type]
9696
origin="start",
9797
closed="left", # type: ignore
9898
).as_expr("datetime_col", "timestamp_col"),
@@ -334,7 +334,7 @@ def test_integer_label_to_datetime_fixed(scalar_types_df: bpd.DataFrame, snapsho
334334
bf_df = scalar_types_df[col_names]
335335
ops_map = {
336336
"fixed_freq": ops.IntegerLabelToDatetimeOp(
337-
freq=pd.tseries.offsets.Day(),
337+
freq=pd.tseries.offsets.Day(), # type: ignore[arg-type]
338338
origin="start",
339339
label="left", # type: ignore
340340
).as_expr("rowindex", "timestamp_col"),
@@ -349,7 +349,7 @@ def test_integer_label_to_datetime_week(scalar_types_df: bpd.DataFrame, snapshot
349349
bf_df = scalar_types_df[col_names]
350350
ops_map = {
351351
"non_fixed_freq_weekly": ops.IntegerLabelToDatetimeOp(
352-
freq=pd.tseries.offsets.Week(weekday=6),
352+
freq=pd.tseries.offsets.Week(weekday=6), # type: ignore[arg-type]
353353
origin="start",
354354
label="left", # type: ignore
355355
).as_expr("rowindex", "timestamp_col"),

0 commit comments

Comments
 (0)