File tree Expand file tree Collapse file tree 2 files changed +7
-2
lines changed
Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -1029,7 +1029,7 @@ def _read_pandas(
10291029 mem_usage = pandas_dataframe .memory_usage (deep = True ).sum ()
10301030 if write_engine == "default" :
10311031 write_engine = (
1032- "bigquery_write"
1032+ bigframes . options . compute . default_write_engine
10331033 if mem_usage > bigframes .constants .MAX_INLINE_BYTES
10341034 else "bigquery_inline"
10351035 )
Original file line number Diff line number Diff line change @@ -87,13 +87,18 @@ def test_progress_bar_extract_jobs(
8787def test_progress_bar_load_jobs (
8888 session : bf .Session , penguins_pandas_df_default_index : pd .DataFrame , capsys
8989):
90+
9091 # repeat the DF to be big enough to trigger the load job.
9192 df = penguins_pandas_df_default_index
9293 while len (df ) < MAX_INLINE_DF_BYTES :
9394 df = pd .DataFrame (np .repeat (df .values , 2 , axis = 0 ))
9495
96+ # default write engine usually streaming, which doesn't have job
9597 with bf .option_context (
96- "display.progress_bar" , "terminal"
98+ "display.progress_bar" ,
99+ "terminal" ,
100+ "compute.default_write_engine" ,
101+ "bigquery_load" ,
97102 ), tempfile .TemporaryDirectory () as dir :
98103 path = dir + "/test_read_csv_progress_bar*.csv"
99104 df .to_csv (path , index = False )
You can’t perform that action at this time.
0 commit comments