Fix lint

Author: ericfe-google

bigquery_magics/bigquery.py

@@ -665,8 +665,12 @@ def _get_graph_schema(
         FROM `{query_job.configuration.destination.project}.{dataset_id}`.INFORMATION_SCHEMA.PROPERTY_GRAPHS
         WHERE PROPERTY_GRAPH_NAME = @graph_id
     """
-    job_config = bigquery.QueryJobConfig(query_parameters=[bigquery.ScalarQueryParameter("graph_id", "STRING", graph_id)])
-    info_schema_results = bq_client.query(info_schema_query, job_config=job_config).to_dataframe()
+    job_config = bigquery.QueryJobConfig(
+        query_parameters=[bigquery.ScalarQueryParameter("graph_id", "STRING", graph_id)]
+    )
+    info_schema_results = bq_client.query(
+        info_schema_query, job_config=job_config
+    ).to_dataframe()
 
     if info_schema_results.shape == (1, 1):
         return graph_server._convert_schema(info_schema_results.iloc[0, 0])

tests/unit/bigquery/test_bigquery.py

@@ -1157,7 +1157,7 @@ def test_add_graph_widget_with_schema(monkeypatch):
         query_str = call_args[0]
         assert "INFORMATION_SCHEMA.PROPERTY_GRAPHS" in query_str
         assert "PROPERTY_GRAPH_NAME = @graph_id" in query_str
-        
+
         # Verify query parameter
         job_config = call_kwargs["job_config"]
         param = job_config.query_parameters[0]