fix lint

Author: antoineeripret

pandas_gbq/gbq_connector.py

@@ -272,7 +272,9 @@ def run_query(
         if dry_run:
             # Access total_bytes_processed from the QueryJob via RowIterator.job
             # RowIterator has a job attribute that references the QueryJob
-            query_job = rows_iter.job if hasattr(rows_iter, 'job') and rows_iter.job else None
+            query_job = (
+                rows_iter.job if hasattr(rows_iter, "job") and rows_iter.job else None
+            )
             if query_job is None:
                 # Fallback: if query_and_wait_via_client_library doesn't set job,
                 # we need to get it from the query result

pandas_gbq/query.py

@@ -182,7 +182,7 @@ def query_and_wait(
         rows_iter = query_reply.result(max_results=max_results)
         # Store reference to QueryJob in RowIterator for dry_run access
         # RowIterator already has a job attribute, but ensure it's set
-        if not hasattr(rows_iter, 'job') or rows_iter.job is None:
+        if not hasattr(rows_iter, "job") or rows_iter.job is None:
             rows_iter.job = query_reply
         return rows_iter
     except connector.http_error as ex:
@@ -217,10 +217,10 @@ def query_and_wait_via_client_library(
         query_job.result(timeout=timeout_ms / 1000.0 if timeout_ms else None)
         # Get the result iterator and ensure job attribute is set
         rows_iter = query_job.result(max_results=max_results)
-        if not hasattr(rows_iter, 'job') or rows_iter.job is None:
+        if not hasattr(rows_iter, "job") or rows_iter.job is None:
             rows_iter.job = query_job
         return rows_iter
-    
+
     rows_iter = try_query(
         connector,
         functools.partial(
@@ -234,7 +234,7 @@ def query_and_wait_via_client_library(
         ),
     )
     # Ensure job attribute is set for consistency
-    if hasattr(rows_iter, 'job') and rows_iter.job is None:
+    if hasattr(rows_iter, "job") and rows_iter.job is None:
         # If query_and_wait doesn't set job, we need to get it from the query
         # This shouldn't happen, but we ensure it's set for dry_run compatibility
         pass

tests/system/test_gbq.py

@@ -669,7 +669,6 @@ def test_read_gbq_with_dry_run(self, project_id):
         assert cost > 0
 
 
-
 class TestToGBQIntegration(object):
     @pytest.fixture(autouse=True, scope="function")
     def setup(self, project, credentials, random_dataset_id):

tests/unit/test_gbq.py

@@ -945,7 +945,10 @@ def test_read_gbq_with_dry_run(mock_bigquery_client, mock_query_job):
     type(mock_query_job).total_bytes_processed = mock.PropertyMock(return_value=12345)
     cost = gbq.read_gbq("SELECT 1", project_id="my-project", dry_run=True)
     # Check which method was called based on BigQuery version
-    if hasattr(mock_bigquery_client, "query_and_wait") and mock_bigquery_client.query_and_wait.called:
+    if (
+        hasattr(mock_bigquery_client, "query_and_wait")
+        and mock_bigquery_client.query_and_wait.called
+    ):
         _, kwargs = mock_bigquery_client.query_and_wait.call_args
         job_config = kwargs["job_config"]
     else: