Skip to content

Commit e38f0c6

Browse files
fix
1 parent d5c861d commit e38f0c6

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

packages/bigframes/bigframes/core/events.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from __future__ import annotations
1616

1717
import asyncio
18+
import concurrent.futures
1819
import dataclasses
1920
import datetime
2021
import threading

packages/bigframes/bigframes/session/execution_spec.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from google.cloud import bigquery
2121

22-
from bigframes.options import ComputeOptions
22+
from bigframes._config import ComputeOptions
2323

2424

2525
@dataclasses.dataclass(frozen=True)
@@ -63,7 +63,7 @@ class ExecutionSpec:
6363

6464
def with_bq_labels(self, labels: Mapping[str, str]) -> ExecutionSpec:
6565
bq_config = self.bigquery_config or BqComputeOptions()
66-
return dataclasses.replace(self, bigquery_config=bq_config.add_labels(labels))
66+
return dataclasses.replace(self, bigquery_config=bq_config.push_labels(labels))
6767

6868
def with_compute_options(self, compute_options: ComputeOptions) -> ExecutionSpec:
6969
"""

0 commit comments

Comments
 (0)