fix seed load result

dbt-labs · Nov 7, 2024 · b8408c2 · b8408c2
1 parent a2db35b
commit b8408c2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/dbt/adapters/bigquery/connections.py b/dbt/adapters/bigquery/connections.py
@@ -496,11 +496,13 @@ def _load_table_from_file(
             with open(file_path, "rb") as f:
                 job = client.load_table_from_file(f, table, rewind=True, job_config=config)
 
-        if not job.done(retry=self._retry.retry(fallback_timeout=fallback_timeout)):
+        response = job.result(retry=self._retry.retry(fallback_timeout=fallback_timeout))
+
+        if response.state != "DONE":
             raise DbtRuntimeError("BigQuery Timeout Exceeded")
 
-        elif job.error_result:
-            message = "\n".join(error["message"].strip() for error in job.errors)
+        elif response.error_result:
+            message = "\n".join(error["message"].strip() for error in response.errors)
             raise DbtRuntimeError(message)
 
     @staticmethod