Skip to content

Commit d84ec94

Browse files
shuoweilshuoweil
committed
style: enforce ruff formatting
1 parent b9476e6 commit d84ec94

File tree

4 files changed

+60
-90
lines changed

4 files changed

+60
-90
lines changed

‎packages/bigframes/.python-version‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
3.14.2

‎packages/bigframes/bigframes/dataframe.py‎

Lines changed: 17 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -1611,8 +1611,7 @@ def to_pandas( # type: ignore[overload-overlap]
16111611
ordered: bool = ...,
16121612
dry_run: Literal[False] = ...,
16131613
allow_large_results: Optional[bool] = ...,
1614-
) -> pandas.DataFrame:
1615-
...
1614+
) -> pandas.DataFrame: ...
16161615

16171616
@overload
16181617
def to_pandas(
@@ -1624,8 +1623,7 @@ def to_pandas(
16241623
ordered: bool = ...,
16251624
dry_run: Literal[True] = ...,
16261625
allow_large_results: Optional[bool] = ...,
1627-
) -> pandas.Series:
1628-
...
1626+
) -> pandas.Series: ...
16291627

16301628
def to_pandas(
16311629
self,
@@ -1937,8 +1935,7 @@ def drop(
19371935
columns: Union[blocks.Label, Sequence[blocks.Label]] = None,
19381936
level: typing.Optional[LevelType] = None,
19391937
inplace: Literal[False] = False,
1940-
) -> DataFrame:
1941-
...
1938+
) -> DataFrame: ...
19421939

19431940
@overload
19441941
def drop(
@@ -1950,8 +1947,7 @@ def drop(
19501947
columns: Union[blocks.Label, Sequence[blocks.Label]] = None,
19511948
level: typing.Optional[LevelType] = None,
19521949
inplace: Literal[True],
1953-
) -> None:
1954-
...
1950+
) -> None: ...
19551951

19561952
def drop(
19571953
self,
@@ -2095,20 +2091,17 @@ def _resolve_levels(self, level: LevelsType) -> typing.Sequence[str]:
20952091
return self._block.index.resolve_level(level)
20962092

20972093
@overload
2098-
def rename(self, *, columns: Mapping[blocks.Label, blocks.Label]) -> DataFrame:
2099-
...
2094+
def rename(self, *, columns: Mapping[blocks.Label, blocks.Label]) -> DataFrame: ...
21002095

21012096
@overload
21022097
def rename(
21032098
self, *, columns: Mapping[blocks.Label, blocks.Label], inplace: Literal[False]
2104-
) -> DataFrame:
2105-
...
2099+
) -> DataFrame: ...
21062100

21072101
@overload
21082102
def rename(
21092103
self, *, columns: Mapping[blocks.Label, blocks.Label], inplace: Literal[True]
2110-
) -> None:
2111-
...
2104+
) -> None: ...
21122105

21132106
def rename(
21142107
self, *, columns: Mapping[blocks.Label, blocks.Label], inplace: bool = False
@@ -2125,8 +2118,7 @@ def rename(
21252118
def rename_axis(
21262119
self,
21272120
mapper: typing.Union[blocks.Label, typing.Sequence[blocks.Label]],
2128-
) -> DataFrame:
2129-
...
2121+
) -> DataFrame: ...
21302122

21312123
@overload
21322124
def rename_axis(
@@ -2135,8 +2127,7 @@ def rename_axis(
21352127
*,
21362128
inplace: Literal[False],
21372129
**kwargs,
2138-
) -> DataFrame:
2139-
...
2130+
) -> DataFrame: ...
21402131

21412132
@overload
21422133
def rename_axis(
@@ -2145,8 +2136,7 @@ def rename_axis(
21452136
*,
21462137
inplace: Literal[True],
21472138
**kwargs,
2148-
) -> None:
2149-
...
2139+
) -> None: ...
21502140

21512141
def rename_axis(
21522142
self,
@@ -2342,8 +2332,7 @@ def reset_index(
23422332
col_fill: Hashable = ...,
23432333
allow_duplicates: Optional[bool] = ...,
23442334
names: Union[None, Hashable, Sequence[Hashable]] = ...,
2345-
) -> DataFrame:
2346-
...
2335+
) -> DataFrame: ...
23472336

23482337
@overload
23492338
def reset_index(
@@ -2355,8 +2344,7 @@ def reset_index(
23552344
col_fill: Hashable = ...,
23562345
allow_duplicates: Optional[bool] = ...,
23572346
names: Union[None, Hashable, Sequence[Hashable]] = ...,
2358-
) -> None:
2359-
...
2347+
) -> None: ...
23602348

23612349
@overload
23622350
def reset_index(
@@ -2368,8 +2356,7 @@ def reset_index(
23682356
col_fill: Hashable = ...,
23692357
allow_duplicates: Optional[bool] = ...,
23702358
names: Union[None, Hashable, Sequence[Hashable]] = ...,
2371-
) -> Optional[DataFrame]:
2372-
...
2359+
) -> Optional[DataFrame]: ...
23732360

23742361
def reset_index(
23752362
self,
@@ -2432,8 +2419,7 @@ def sort_index(
24322419
ascending: bool = ...,
24332420
inplace: Literal[False] = ...,
24342421
na_position: Literal["first", "last"] = ...,
2435-
) -> DataFrame:
2436-
...
2422+
) -> DataFrame: ...
24372423

24382424
@overload
24392425
def sort_index(
@@ -2442,8 +2428,7 @@ def sort_index(
24422428
ascending: bool = ...,
24432429
inplace: Literal[True] = ...,
24442430
na_position: Literal["first", "last"] = ...,
2445-
) -> None:
2446-
...
2431+
) -> None: ...
24472432

24482433
def sort_index(
24492434
self,
@@ -2489,8 +2474,7 @@ def sort_values(
24892474
ascending: bool | typing.Sequence[bool] = ...,
24902475
kind: str = ...,
24912476
na_position: typing.Literal["first", "last"] = ...,
2492-
) -> DataFrame:
2493-
...
2477+
) -> DataFrame: ...
24942478

24952479
@overload
24962480
def sort_values(
@@ -2501,8 +2485,7 @@ def sort_values(
25012485
ascending: bool | typing.Sequence[bool] = ...,
25022486
kind: str = ...,
25032487
na_position: typing.Literal["first", "last"] = ...,
2504-
) -> None:
2505-
...
2488+
) -> None: ...
25062489

25072490
def sort_values(
25082491
self,

‎packages/bigframes/bigframes/series.py‎

Lines changed: 31 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -375,8 +375,7 @@ def copy(self) -> Series:
375375
def rename(
376376
self,
377377
index: Union[blocks.Label, Mapping[Any, Any]] = None,
378-
) -> Series:
379-
...
378+
) -> Series: ...
380379

381380
@overload
382381
def rename(
@@ -385,8 +384,7 @@ def rename(
385384
*,
386385
inplace: Literal[False],
387386
**kwargs,
388-
) -> Series:
389-
...
387+
) -> Series: ...
390388

391389
@overload
392390
def rename(
@@ -395,8 +393,7 @@ def rename(
395393
*,
396394
inplace: Literal[True],
397395
**kwargs,
398-
) -> None:
399-
...
396+
) -> None: ...
400397

401398
def rename(
402399
self,
@@ -457,8 +454,7 @@ def rename(
457454
def rename_axis(
458455
self,
459456
mapper: typing.Union[blocks.Label, typing.Sequence[blocks.Label]],
460-
) -> Series:
461-
...
457+
) -> Series: ...
462458

463459
@overload
464460
def rename_axis(
@@ -467,8 +463,7 @@ def rename_axis(
467463
*,
468464
inplace: Literal[False],
469465
**kwargs,
470-
) -> Series:
471-
...
466+
) -> Series: ...
472467

473468
@overload
474469
def rename_axis(
@@ -477,8 +472,7 @@ def rename_axis(
477472
*,
478473
inplace: Literal[True],
479474
**kwargs,
480-
) -> None:
481-
...
475+
) -> None: ...
482476

483477
@validations.requires_index
484478
def rename_axis(
@@ -522,8 +516,7 @@ def reset_index(
522516
drop: Literal[False] = ...,
523517
inplace: Literal[False] = ...,
524518
allow_duplicates: Optional[bool] = ...,
525-
) -> bigframes.dataframe.DataFrame:
526-
...
519+
) -> bigframes.dataframe.DataFrame: ...
527520

528521
@overload
529522
def reset_index(
@@ -534,8 +527,7 @@ def reset_index(
534527
drop: Literal[True] = ...,
535528
inplace: Literal[False] = ...,
536529
allow_duplicates: Optional[bool] = ...,
537-
) -> Series:
538-
...
530+
) -> Series: ...
539531

540532
@overload
541533
def reset_index(
@@ -546,8 +538,7 @@ def reset_index(
546538
drop: bool = ...,
547539
inplace: Literal[True] = ...,
548540
allow_duplicates: Optional[bool] = ...,
549-
) -> None:
550-
...
541+
) -> None: ...
551542

552543
@validations.requires_ordering()
553544
def reset_index(
@@ -1540,9 +1531,9 @@ def ne(self, other: object) -> Series:
15401531

15411532
def items(self):
15421533
for batch_df in self._block.to_pandas_batches():
1543-
assert (
1544-
batch_df.shape[1] == 1
1545-
), f"Expected 1 column in the dataframe, but got {batch_df.shape[1]}."
1534+
assert batch_df.shape[1] == 1, (
1535+
f"Expected 1 column in the dataframe, but got {batch_df.shape[1]}."
1536+
)
15461537
for item in batch_df.squeeze(axis=1).items():
15471538
yield item
15481539

@@ -1772,8 +1763,7 @@ def sort_values(
17721763
ascending: bool | typing.Sequence[bool] = ...,
17731764
kind: str = ...,
17741765
na_position: typing.Literal["first", "last"] = ...,
1775-
) -> None:
1776-
...
1766+
) -> None: ...
17771767

17781768
@typing.overload
17791769
def sort_values(
@@ -1784,8 +1774,7 @@ def sort_values(
17841774
ascending: bool | typing.Sequence[bool] = ...,
17851775
kind: str = ...,
17861776
na_position: typing.Literal["first", "last"] = ...,
1787-
) -> Series:
1788-
...
1777+
) -> Series: ...
17891778

17901779
def sort_values(
17911780
self,
@@ -1816,14 +1805,12 @@ def sort_values(
18161805
@typing.overload # type: ignore[override]
18171806
def sort_index(
18181807
self, *, axis=..., inplace: Literal[False] = ..., ascending=..., na_position=...
1819-
) -> Series:
1820-
...
1808+
) -> Series: ...
18211809

18221810
@typing.overload
18231811
def sort_index(
18241812
self, *, axis=0, inplace: Literal[True] = ..., ascending=..., na_position=...
1825-
) -> None:
1826-
...
1813+
) -> None: ...
18271814

18281815
@validations.requires_index
18291816
def sort_index(
@@ -2698,18 +2685,28 @@ def _apply_binary_aggregation(
26982685
@typing.overload
26992686
def _align(
27002687
self, other: Series, how="outer"
2701-
) -> tuple[ex.DerefOp, ex.DerefOp, blocks.Block,]:
2702-
...
2688+
) -> tuple[
2689+
ex.DerefOp,
2690+
ex.DerefOp,
2691+
blocks.Block,
2692+
]: ...
27032693

27042694
@typing.overload
27052695
def _align(
27062696
self, other: typing.Union[Series, scalars.Scalar], how="outer"
2707-
) -> tuple[ex.DerefOp, AlignedExprT, blocks.Block,]:
2708-
...
2697+
) -> tuple[
2698+
ex.DerefOp,
2699+
AlignedExprT,
2700+
blocks.Block,
2701+
]: ...
27092702

27102703
def _align(
27112704
self, other: typing.Union[Series, scalars.Scalar], how="outer"
2712-
) -> tuple[ex.DerefOp, AlignedExprT, blocks.Block,]:
2705+
) -> tuple[
2706+
ex.DerefOp,
2707+
AlignedExprT,
2708+
blocks.Block,
2709+
]:
27132710
"""Aligns the series value with another scalar or series object. Returns new left column id, right column id and joined tabled expression."""
27142711
values, block = self._align_n(
27152712
[

0 commit comments

Comments
 (0)