diff --git a/bigframes/bigquery/__init__.py b/bigframes/bigquery/__init__.py index 5728f153dd..150fe5efc0 100644 --- a/bigframes/bigquery/__init__.py +++ b/bigframes/bigquery/__init__.py @@ -60,7 +60,7 @@ from bigframes.bigquery._operations.search import create_vector_index, vector_search from bigframes.bigquery._operations.sql import sql_scalar from bigframes.bigquery._operations.struct import struct -from bigframes.bigquery.table import create_external_table +from bigframes.bigquery._operations.table import create_external_table from bigframes.core.logging import log_adapter _functions = [ diff --git a/bigframes/bigquery/table.py b/bigframes/bigquery/_operations/table.py similarity index 100% rename from bigframes/bigquery/table.py rename to bigframes/bigquery/_operations/table.py diff --git a/tests/unit/bigquery/test_table.py b/tests/unit/bigquery/test_table.py index 441130d53d..badce5e5e2 100644 --- a/tests/unit/bigquery/test_table.py +++ b/tests/unit/bigquery/test_table.py @@ -16,7 +16,7 @@ import pytest -import bigframes.bigquery.table +import bigframes.bigquery import bigframes.core.sql.table import bigframes.session @@ -80,9 +80,9 @@ def test_create_external_table_ddl_connection(): assert sql == expected -@mock.patch("bigframes.bigquery.table._get_table_metadata") +@mock.patch("bigframes.bigquery._operations.table._get_table_metadata") def test_create_external_table(get_table_metadata_mock, mock_session): - bigframes.bigquery.table.create_external_table( + bigframes.bigquery.create_external_table( "my-project.my_dataset.my_table", columns={"col1": "INT64", "col2": "STRING"}, options={"format": "CSV", "uris": ["gs://bucket/path*"]},