Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit c9eaff0

Browse files
authoredJul 16, 2024
feat: Add bigframes-mode label to query jobs (#832)
1 parent 6278e0b commit c9eaff0

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed
 

‎bigframes/session/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -1849,6 +1849,8 @@ def _start_query(
18491849
Starts BigQuery query job and waits for results.
18501850
"""
18511851
job_config = self._prepare_query_job_config(job_config)
1852+
if not self._strictly_ordered:
1853+
job_config.labels = {"bigframes-mode": "unordered"}
18521854
try:
18531855
return bigframes.session._io.bigquery.start_query_with_client(
18541856
self,

‎tests/system/small/test_unordered.py

+9
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,15 @@
2020
from tests.system.utils import assert_pandas_df_equal, skip_legacy_pandas
2121

2222

23+
def test_unordered_mode_job_label(unordered_session):
24+
pd_df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}, dtype=pd.Int64Dtype())
25+
df = bpd.DataFrame(pd_df, session=unordered_session)
26+
df.to_pandas()
27+
job_labels = df.query_job.labels # type:ignore
28+
assert "bigframes-mode" in job_labels
29+
assert job_labels["bigframes-mode"] == "unordered"
30+
31+
2332
def test_unordered_mode_cache_aggregate(unordered_session):
2433
pd_df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}, dtype=pd.Int64Dtype())
2534
df = bpd.DataFrame(pd_df, session=unordered_session)

0 commit comments

Comments
 (0)
Failed to load comments.