query_job fix

dlt-hub · Mar 28, 2024 · 4191191 · 4191191
1 parent 0ff0ea6
commit 4191191
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlt/destinations/impl/bigquery/sql_client.py b/dlt/destinations/impl/bigquery/sql_client.py
@@ -48,7 +48,7 @@ class BigQueryDBApiCursorImpl(DBApiCursorImpl):
     def df(self, chunk_size: int = None, **kwargs: Any) -> DataFrame:
         if chunk_size is not None:
             return super().df(chunk_size=chunk_size)
-        query_job: bigquery.QueryJob = self.native_cursor._query_job  # type: ignore
+        query_job: bigquery.QueryJob = getattr(self.native_cursor, "_query_job", self.native_cursor.query_job)  # type: ignore
 
         try:
             return query_job.to_dataframe(**kwargs)