Skip to content

Commit 7e02454

Browse files
committed
Revert "fix: Ensure _read_gbq_colab correctly logs BigQuery job labels"
This reverts commit 5c018ac.
1 parent 5c018ac commit 7e02454

File tree

1 file changed

+0
-15
lines changed

1 file changed

+0
-15
lines changed

bigframes/testing/mocks.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,7 @@
2727
import bigframes
2828
import bigframes.clients
2929
import bigframes.core.global_session
30-
import bigframes.core.log_adapter
3130
import bigframes.dataframe
32-
import bigframes.session._io.bigquery
33-
from bigframes.session._io.bigquery import create_job_configs_labels
3431
import bigframes.session.clients
3532

3633
"""Utilities for creating test resources."""
@@ -93,18 +90,6 @@ def query_mock(
9390
job_config: Optional[google.cloud.bigquery.QueryJobConfig] = None,
9491
**kwargs,
9592
):
96-
job_config = (
97-
job_config
98-
if job_config is not None
99-
else google.cloud.bigquery.QueryJobConfig()
100-
)
101-
api_methods = bigframes.core.log_adapter.get_and_reset_api_methods(
102-
dry_run=job_config.dry_run
103-
)
104-
job_config.labels = create_job_configs_labels(
105-
job_configs_labels=job_config.labels,
106-
api_methods=api_methods,
107-
)
10893
queries.append(query)
10994
job_configs.append(copy.deepcopy(job_config))
11095
query_job = mock.create_autospec(google.cloud.bigquery.QueryJob, instance=True)

0 commit comments

Comments
 (0)