Merge branch 'main' into salem_query_externalsheetspermanent_update

googleapis · Jan 26, 2024 · 46ef13d · 46ef13d
2 parents da3ea0e + dbf10de
commit 46ef13d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/samples/client_query_destination_table_legacy.py b/samples/client_query_destination_table_legacy.py
@@ -36,8 +36,9 @@ def client_query_destination_table_legacy(table_id: str) -> None:
  """
 
  # Start the query, passing in the extra configuration.
- query_job = client.query(sql, job_config=job_config) # Make an API request.
- query_job.result() # Wait for the job to complete.
+ client.query_and_wait(
+ sql, job_config=job_config
+ ) # Make an API request and wait for the query to finish.
 
  print("Query results loaded to the table {}".format(table_id))
  # [END bigquery_query_legacy_large_results]
diff --git a/samples/client_query_w_positional_params.py b/samples/client_query_w_positional_params.py
@@ -35,8 +35,10 @@ def client_query_w_positional_params() -> None:
  bigquery.ScalarQueryParameter(None, "INT64", 250),
  ]
  )
- query_job = client.query(query, job_config=job_config) # Make an API request.
+ results = client.query_and_wait(
+ query, job_config=job_config
+ ) # Make an API request.
 
- for row in query_job:
+ for row in results:
  print("{}: \t{}".format(row.word, row.word_count))
  # [END bigquery_query_params_positional]
diff --git a/samples/query_to_arrow.py b/samples/query_to_arrow.py
@@ -44,8 +44,8 @@ def query_to_arrow() -> "pyarrow.Table":
  FROM races r
  CROSS JOIN UNNEST(r.participants) as participant;
  """
- query_job = client.query(sql)
- arrow_table = query_job.to_arrow() # Make an API request.
+ results = client.query_and_wait(sql)
+ arrow_table = results.to_arrow() # Make an API request.
 
  print(
  "Downloaded {} rows, {} columns.".format(