Skip to content

Commit 11af385

Browse files
authored
chore: overload update for pre-commit mypy. (#773)
1 parent 91efe8e commit 11af385

File tree

3 files changed

+10
-11
lines changed

3 files changed

+10
-11
lines changed

bigframes/core/tools/datetimes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929
def to_datetime(
3030
arg: Union[
31-
vendored_pandas_datetimes.local_scalars,
31+
Union[int, float, str, datetime],
3232
vendored_pandas_datetimes.local_iterables,
3333
bigframes.series.Series,
3434
bigframes.dataframe.DataFrame,

bigframes/pandas/__init__.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -701,33 +701,33 @@ def read_gbq_function(function_name: str):
701701

702702
@typing.overload
703703
def to_datetime(
704-
arg: vendored_pandas_datetimes.local_scalars,
704+
arg: Union[
705+
vendored_pandas_datetimes.local_iterables,
706+
bigframes.series.Series,
707+
bigframes.dataframe.DataFrame,
708+
],
705709
*,
706710
utc: bool = False,
707711
format: Optional[str] = None,
708712
unit: Optional[str] = None,
709-
) -> Union[pandas.Timestamp, datetime]:
713+
) -> bigframes.series.Series:
710714
...
711715

712716

713717
@typing.overload
714718
def to_datetime(
715-
arg: Union[
716-
vendored_pandas_datetimes.local_iterables,
717-
bigframes.series.Series,
718-
bigframes.dataframe.DataFrame,
719-
],
719+
arg: Union[int, float, str, datetime],
720720
*,
721721
utc: bool = False,
722722
format: Optional[str] = None,
723723
unit: Optional[str] = None,
724-
) -> bigframes.series.Series:
724+
) -> Union[pandas.Timestamp, datetime]:
725725
...
726726

727727

728728
def to_datetime(
729729
arg: Union[
730-
vendored_pandas_datetimes.local_scalars,
730+
Union[int, float, str, datetime],
731731
vendored_pandas_datetimes.local_iterables,
732732
bigframes.series.Series,
733733
bigframes.dataframe.DataFrame,

third_party/bigframes_vendored/pandas/core/tools/datetimes.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77

88
from bigframes import constants, series
99

10-
local_scalars = Union[int, float, str, datetime]
1110
local_iterables = Union[List, Tuple, pd.Series, pd.DataFrame, Mapping]
1211

1312

0 commit comments

Comments
 (0)