chore: lint

Author: tswast

pandas_gbq/core/biglake.py

@@ -30,6 +30,7 @@
 FROM `{project}.{catalog}.{namespace}.{table}`
 """
 
+
 @dataclasses.dataclass(frozen=True)
 class BigLakeTableMetadata:
     schema: Sequence[google.cloud.bigquery.SchemaField]
@@ -59,17 +60,22 @@ def get_table_metadata(
     job.result()
     schema = job.schema
 
-    count_rows = list(bqclient.query_and_wait(_COUNT_TEMPLATE.format(
-        project=reference.project,
-        catalog=reference.catalog,
-        namespace=".".join(reference.namespace),
-        table=reference.table,
-    )))
-    assert len(count_rows) == 1, "got unexpected query response when determining number of rows"
+    count_rows = list(
+        bqclient.query_and_wait(
+            _COUNT_TEMPLATE.format(
+                project=reference.project,
+                catalog=reference.catalog,
+                namespace=".".join(reference.namespace),
+                table=reference.table,
+            )
+        )
+    )
+    assert (
+        len(count_rows) == 1
+    ), "got unexpected query response when determining number of rows"
     total_rows = count_rows[0].total_rows
 
     return BigLakeTableMetadata(
         schema=schema if schema is not None else [],
         num_rows=total_rows,
     )
-

pandas_gbq/core/sample.py

@@ -250,12 +250,14 @@ def _sample_bq_table(
     progress_bar_type: str | None,
     use_bqstorage_api: bool,
 ) -> Optional[pandas.DataFrame]:
-    table = bqclient.get_table(google.cloud.bigquery.TableReference(
-        google.cloud.bigquery.DatasetReference(
-            reference.project_id, reference.dataset_id
-        ),
-        reference.table_id
-    ))
+    table = bqclient.get_table(
+        google.cloud.bigquery.TableReference(
+            google.cloud.bigquery.DatasetReference(
+                reference.project_id, reference.dataset_id
+            ),
+            reference.table_id,
+        )
+    )
     num_rows = table.num_rows
     num_bytes = table.num_bytes
     table_type = table.table_type

tests/unit/core/test_core_sample.py

@@ -237,7 +237,9 @@ def test_download_results_in_parallel_no_table(
 
 
 @mock.patch("pandas_gbq.core.sample._download_results_in_parallel")
-def test_sample_with_tablesample(mock_download_results_in_parallel, mock_bigquery_client):
+def test_sample_with_tablesample(
+    mock_download_results_in_parallel, mock_bigquery_client
+):
     proportion = 0.1
     target_row_count = 100
 

tests/unit/test_to_gbq.py

@@ -244,8 +244,7 @@ def test_create_user_agent_vscode_plugin():
         with mock.patch("pathlib.Path.home", return_value=user_home):
             result = create_user_agent()
             assert (
-                f"pandas-{pd.__version__} vscode googlecloudtools.cloudcode"
-                in result
+                f"pandas-{pd.__version__} vscode googlecloudtools.cloudcode" in result
             )