diff --git a/packages/bigframes/bigframes/operations/__init__.py b/packages/bigframes/bigframes/operations/__init__.py index 5e0a23005a24..dd036bec5a26 100644 --- a/packages/bigframes/bigframes/operations/__init__.py +++ b/packages/bigframes/bigframes/operations/__init__.py @@ -126,6 +126,7 @@ geo_x_op, geo_y_op, ) +from bigframes.operations.googlesql import GoogleSqlScalarOp from bigframes.operations.json_ops import ( JSONExtract, JSONExtractArray, @@ -229,7 +230,6 @@ timestamp_add_op, timestamp_sub_op, ) -from bigframes.operations.googlesql import GoogleSqlScalarOp __all__ = [ # Base ops diff --git a/packages/bigframes/bigframes/operations/geo_ops.py b/packages/bigframes/bigframes/operations/geo_ops.py index 72f3ab48de5b..a965ddca2b9c 100644 --- a/packages/bigframes/bigframes/operations/geo_ops.py +++ b/packages/bigframes/bigframes/operations/geo_ops.py @@ -19,7 +19,6 @@ from bigframes import dtypes from bigframes.operations import base_ops - GeoStAstextOp = base_ops.create_unary_op( name="geo_st_astext", type_signature=op_typing.FixedOutputType( diff --git a/packages/bigframes/bigframes/testing/engine_utils.py b/packages/bigframes/bigframes/testing/engine_utils.py index 456d6b463840..fde713ad2bb5 100644 --- a/packages/bigframes/bigframes/testing/engine_utils.py +++ b/packages/bigframes/bigframes/testing/engine_utils.py @@ -15,8 +15,7 @@ import pandas.testing from bigframes.core import nodes -from bigframes.session import semi_executor, execution_spec - +from bigframes.session import execution_spec, semi_executor SPEC = execution_spec.ExecutionSpec( ordered=True, diff --git a/packages/bigframes/tests/system/conftest.py b/packages/bigframes/tests/system/conftest.py index 8c6de23c9bd3..1476d41325dc 100644 --- a/packages/bigframes/tests/system/conftest.py +++ b/packages/bigframes/tests/system/conftest.py @@ -27,9 +27,9 @@ import google.api_core.exceptions import google.cloud.bigquery as bigquery import google.cloud.bigquery_connection_v1 as bigquery_connection_v1 +import google.cloud.bigquery_storage_v1 import google.cloud.exceptions import google.cloud.functions_v2 as functions_v2 -import google.cloud.bigquery_storage_v1 import google.cloud.resourcemanager_v3 as resourcemanager_v3 import google.cloud.storage as storage # type: ignore import numpy as np diff --git a/packages/bigframes/tests/system/small/engines/test_sorting.py b/packages/bigframes/tests/system/small/engines/test_sorting.py index bcab19f324dd..6593e0154b9a 100644 --- a/packages/bigframes/tests/system/small/engines/test_sorting.py +++ b/packages/bigframes/tests/system/small/engines/test_sorting.py @@ -16,7 +16,7 @@ import bigframes.operations as bf_ops from bigframes.core import array_value, nodes, ordering -from bigframes.session import polars_executor, execution_spec +from bigframes.session import execution_spec, polars_executor from bigframes.testing.engine_utils import assert_equivalence_execution pytest.importorskip("polars")