chore: reformat using ruff

Author: chalmerlowe

packages/bigframes/bigframes/_config/auth.py

@@ -30,9 +30,9 @@
 _cached_project_default: Optional[str] = None
 
 
-def get_default_credentials_with_project() -> (
-    tuple[google.auth.credentials.Credentials, Optional[str]]
-):
+def get_default_credentials_with_project() -> tuple[
+    google.auth.credentials.Credentials, Optional[str]
+]:
     global _AUTH_LOCK, _cached_credentials, _cached_project_default
 
     with _AUTH_LOCK:

packages/bigframes/bigframes/bigquery/__init__.py

@@ -14,7 +14,7 @@
 
 """This module integrates BigQuery built-in functions for use with DataFrame objects,
 such as array functions:
-https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions. """
+https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions."""
 
 import sys
 

packages/bigframes/bigframes/bigquery/_operations/array.py

@@ -17,7 +17,6 @@
 https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions
 """
 
-
 from __future__ import annotations
 
 import typing

packages/bigframes/bigframes/bigquery/_operations/json.py

@@ -18,7 +18,6 @@
 https://cloud.google.com/bigquery/docs/reference/standard-sql/json_functions
 """
 
-
 from __future__ import annotations
 
 import warnings

packages/bigframes/bigframes/bigquery/_operations/ml.py

@@ -131,9 +131,9 @@ def create_model(
     if session is None:
         bpd.read_gbq_query(sql)
         session = bpd.get_global_session()
-        assert (
-            session is not None
-        ), f"Missing connection to BigQuery. Please report how you encountered this error at {bigframes_vendored.constants.FEEDBACK_LINK}."
+        assert session is not None, (
+            f"Missing connection to BigQuery. Please report how you encountered this error at {bigframes_vendored.constants.FEEDBACK_LINK}."
+        )
     else:
         session.read_gbq_query(sql)
 

packages/bigframes/bigframes/bigquery/_operations/obj.py

@@ -18,7 +18,6 @@
 See bigframes.bigquery.obj for public docs.
 """
 
-
 from __future__ import annotations
 
 import datetime

packages/bigframes/bigframes/bigquery/_operations/struct.py

@@ -15,8 +15,7 @@
 
 """This module integrates BigQuery built-in functions for use with DataFrame objects,
 such as array functions:
-https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions. """
-
+https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions."""
 
 from __future__ import annotations
 

packages/bigframes/bigframes/core/agg_expressions.py

@@ -60,8 +60,7 @@ def output_type(self) -> dtypes.ExpressionType:
     @abc.abstractmethod
     def inputs(
         self,
-    ) -> typing.Tuple[expression.Expression, ...]:
-        ...
+    ) -> typing.Tuple[expression.Expression, ...]: ...
 
     @property
     def children(self) -> Tuple[expression.Expression, ...]:
@@ -82,8 +81,7 @@ def is_scalar_expr(self) -> bool:
         return False
 
     @abc.abstractmethod
-    def replace_args(self: TExpression, *arg) -> TExpression:
-        ...
+    def replace_args(self: TExpression, *arg) -> TExpression: ...
 
     def transform_children(
         self: TExpression, t: Callable[[expression.Expression], expression.Expression]

packages/bigframes/bigframes/core/bigframe_node.py

@@ -142,8 +142,7 @@ def roots(self) -> typing.Set[BigFrameNode]:
     # TODO: Store some local data lazily for select, aggregate nodes.
     @property
     @abc.abstractmethod
-    def fields(self) -> Sequence[field.Field]:
-        ...
+    def fields(self) -> Sequence[field.Field]: ...
 
     @property
     def ids(self) -> Iterable[identifiers.ColumnId]:
@@ -300,9 +299,9 @@ def iter_nodes_topo(
         self: BigFrameNode,
     ) -> Generator[BigFrameNode, None, None]:
         """Returns nodes in reverse topological order, using Kahn's algorithm."""
-        child_to_parents: Dict[
-            BigFrameNode, list[BigFrameNode]
-        ] = collections.defaultdict(list)
+        child_to_parents: Dict[BigFrameNode, list[BigFrameNode]] = (
+            collections.defaultdict(list)
+        )
         out_degree: Dict[BigFrameNode, int] = collections.defaultdict(int)
 
         queue: collections.deque["BigFrameNode"] = collections.deque()

packages/bigframes/bigframes/core/blocks.py

@@ -2545,7 +2545,10 @@ def join(
         sort: bool = False,
         block_identity_join: bool = False,
         always_order: bool = False,
-    ) -> Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]:
+    ) -> Tuple[
+        Block,
+        Tuple[Mapping[str, str], Mapping[str, str]],
+    ]:
         """
         Join two blocks objects together, and provide mappings between source columns and output columns.
 
@@ -2794,7 +2797,9 @@ def _is_monotonic(
                 block = block.drop_columns([equal_monotonic_id, strict_monotonic_id])
 
         block, monotonic_result_id = block.apply_binary_op(
-            last_result_id, last_notna_id, ops.and_op  # type: ignore
+            last_result_id,
+            last_notna_id,
+            ops.and_op,  # type: ignore
         )
         if last_result_id is not None:
             block = block.drop_columns([last_result_id, last_notna_id])
@@ -2963,7 +2968,12 @@ def is_uniquely_named(self: BlockIndexProperties):
 
 def try_new_row_join(
     left: Block, right: Block
-) -> Optional[Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]]:
+) -> Optional[
+    Tuple[
+        Block,
+        Tuple[Mapping[str, str], Mapping[str, str]],
+    ]
+]:
     join_keys = tuple(
         (left_id, right_id)
         for left_id, right_id in zip(left.index_columns, right.index_columns)
@@ -2994,7 +3004,12 @@ def try_legacy_row_join(
     right: Block,
     *,
     how="left",
-) -> Optional[Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]]:
+) -> Optional[
+    Tuple[
+        Block,
+        Tuple[Mapping[str, str], Mapping[str, str]],
+    ]
+]:
     """Joins two blocks that have a common root expression by merging the projections."""
     left_expr = left.expr
     right_expr = right.expr
@@ -3048,7 +3063,10 @@ def try_legacy_row_join(
 def join_with_single_row(
     left: Block,
     single_row_block: Block,
-) -> Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]:
+) -> Tuple[
+    Block,
+    Tuple[Mapping[str, str], Mapping[str, str]],
+]:
     """
     Special join case where other is a single row block.
     This property is not validated, caller responsible for not passing multi-row block.
@@ -3083,7 +3101,10 @@ def join_mono_indexed(
     how="left",
     sort: bool = False,
     propogate_order: bool = False,
-) -> Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]:
+) -> Tuple[
+    Block,
+    Tuple[Mapping[str, str], Mapping[str, str]],
+]:
     left_expr = left.expr
     right_expr = right.expr
 
@@ -3130,7 +3151,10 @@ def join_multi_indexed(
     how="left",
     sort: bool = False,
     propogate_order: bool = False,
-) -> Tuple[Block, Tuple[Mapping[str, str], Mapping[str, str]],]:
+) -> Tuple[
+    Block,
+    Tuple[Mapping[str, str], Mapping[str, str]],
+]:
     if not (left.index.is_uniquely_named() and right.index.is_uniquely_named()):
         raise ValueError("Joins not supported on indices with non-unique level names")