Skip to content

Commit 1f8c125

Browse files
committed
format
1 parent 3fe08c2 commit 1f8c125

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

src/pyarrow/bigquery/read.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,7 @@ def read_table(
213213
)
214214
)
215215

216+
216217
def read_query(
217218
project: str,
218219
query: str,
@@ -223,10 +224,6 @@ def read_query(
223224
):
224225
return pa.concat_tables(
225226
reader_query(
226-
project=project,
227-
query=query,
228-
worker_count=worker_count,
229-
worker_type=worker_type,
230-
batch_size=batch_size
227+
project=project, query=query, worker_count=worker_count, worker_type=worker_type, batch_size=batch_size
231228
)
232229
)

src/pyarrow/bigquery/write/upload.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,7 @@
66
from . import pa_to_pb
77

88

9-
@tenacity.retry(
10-
stop=tenacity.stop_after_attempt(5),
11-
retry=tenacity.retry_if_exception_type(Unknown))
9+
@tenacity.retry(stop=tenacity.stop_after_attempt(5), retry=tenacity.retry_if_exception_type(Unknown))
1210
def _send(stream, serialized_rows, offset):
1311
proto_rows = types.ProtoRows()
1412
proto_rows.serialized_rows.extend(serialized_rows)

0 commit comments

Comments
 (0)