chore: updates linting

Author: chalmerlowe

packages/bigframes/bigframes/core/compile/ibis_compiler/operations/geo_ops.py

@@ -182,7 +182,9 @@ def st_buffer(
 
 @ibis_udf.scalar.builtin
 def st_distance(
-    a: ibis_dtypes.geography, b: ibis_dtypes.geography, use_spheroid: bool  # type: ignore
+    a: ibis_dtypes.geography,
+    b: ibis_dtypes.geography,
+    use_spheroid: bool,  # type: ignore
 ) -> ibis_dtypes.float:  # type: ignore
     """Convert string to geography."""
 

packages/bigframes/bigframes/core/compile/ibis_compiler/scalar_op_registry.py

@@ -2172,7 +2172,7 @@ def obj_make_ref_json(objectref_json: ibis_dtypes.JSON) -> _OBJ_REF_IBIS_DTYPE:
 # _OBJ_REF_IBIS_DTYPE is a variable holding a type, Mypy complains about it being used as type hint.
 def obj_get_access_url(  # type: ignore[empty-body]
     obj_ref: _OBJ_REF_IBIS_DTYPE,  # type: ignore[valid-type]
-    mode: ibis_dtypes.String
+    mode: ibis_dtypes.String,
 ) -> ibis_dtypes.JSON:
     """Get access url (as ObjectRefRumtime JSON) from ObjectRef."""
 

packages/bigframes/bigframes/core/compile/sqlglot/compiler.py

@@ -54,7 +54,9 @@ def compile_sql(request: configs.CompileRequest) -> configs.CompileResult:
         # Need to do this before replacing unsupported ops, as that will rewrite slice ops
         result_node = rewrite.pull_up_limits(result_node)
     result_node = typing.cast(nodes.ResultNode, _replace_unsupported_ops(result_node))
-    result_node = typing.cast(nodes.ResultNode, result_node.bottom_up(rewrite.simplify_join))
+    result_node = typing.cast(
+        nodes.ResultNode, result_node.bottom_up(rewrite.simplify_join)
+    )
     # prune before pulling up order to avoid unnnecessary row_number() ops
     result_node = typing.cast(nodes.ResultNode, rewrite.column_pruning(result_node))
     result_node = rewrite.defer_order(

packages/bigframes/bigframes/core/expression_factoring.py

@@ -243,7 +243,12 @@ def factor_aggregation(root: nodes.ColumnDef) -> FactoredAggregation:
     }
 
     root_scalar_expr = nodes.ColumnDef(
-        sub_expressions(root.expression, cast(Mapping[expression.Expression, expression.Expression], agg_outputs_dict)),
+        sub_expressions(
+            root.expression,
+            cast(
+                Mapping[expression.Expression, expression.Expression], agg_outputs_dict
+            ),
+        ),
         root.id,  # type: ignore
     )
 

packages/bigframes/bigframes/core/local_data.py

@@ -33,8 +33,7 @@
 
 import bigframes.core.schema as schemata
 import bigframes.dtypes
-from bigframes.core import identifiers
-from bigframes.core import pyarrow_utils
+from bigframes.core import identifiers, pyarrow_utils
 
 
 @dataclasses.dataclass(frozen=True)

packages/bigframes/bigframes/core/nodes.py

@@ -674,7 +674,9 @@ def fields(self) -> Sequence[Field]:
             Field(
                 col_id,
                 self.local_data_source.schema.get_type(source_id),
-                nullable=self.local_data_source.is_nullable(identifiers.ColumnId(source_id)),
+                nullable=self.local_data_source.is_nullable(
+                    identifiers.ColumnId(source_id)
+                ),
             )
             for col_id, source_id in self.scan_list.items
         )

packages/bigframes/noxfile.py

@@ -364,7 +364,6 @@ def run_system(
     )
 
 
-
 @nox.session(python=SYSTEM_TEST_PYTHON_VERSIONS)
 def system(session: nox.sessions.Session):
     """Run the system test suite."""
@@ -982,7 +981,7 @@ def core_deps_from_source(session, protobuf_implementation):
     # added to the list below so that it is installed from source, rather than PyPI
     # Note: If a dependency is added to the `core_dependencies_from_source` list,
     # the `prerel_deps` list in the `prerelease_deps` nox session should also be updated.
-    core_dependencies_from_source = [        
+    core_dependencies_from_source = [
         "googleapis-common-protos @ git+https://github.com/googleapis/google-cloud-python#egg=googleapis-common-protos&subdirectory=packages/googleapis-common-protos",
         "google-api-core @ git+https://github.com/googleapis/google-cloud-python#egg=google-api-core&subdirectory=packages/google-api-core",
         "google-auth @ git+https://github.com/googleapis/google-cloud-python#egg=google-auth&subdirectory=packages/google-auth",

packages/bigframes/tests/unit/core/compile/sqlglot/tpch/conftest.py

@@ -158,7 +158,7 @@ def read_gbq_table_no_snapshot(*args, **kwargs):
         return original_read_gbq_table(*args, **kwargs)
 
     session._executor = compiler_session.SQLCompilerExecutor()
-    
+
     with mock.patch.object(
         session._loader, "read_gbq_table", new=read_gbq_table_no_snapshot
     ):

packages/bigframes/tests/unit/test_col.py

@@ -16,13 +16,13 @@
 import pathlib
 from typing import Generator
 
+import numpy as np
 import pandas as pd
 import pytest
 
 import bigframes
 import bigframes.pandas as bpd
 from bigframes.testing.utils import assert_frame_equal, convert_pandas_dtypes
-import numpy as np
 
 pytest.importorskip("polars")
 pytest.importorskip("pandas", minversion="3.0.0")