Skip to content

Commit b510cec

Browse files
committed
Fix: Update PyArrow serialization in append_rows_with_arrow.py
1 parent 2239db8 commit b510cec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

packages/google-cloud-bigquery-storage/samples/pyarrow/append_rows_with_arrow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def verify_result(client, table, futures):
219219
query = client.query(f"SELECT DISTINCT int64_col FROM `{bq_table}`;")
220220
query_result = query.result().to_dataframe()
221221

222-
assert query_result.iloc[0, 0] == TABLE_LENGTH
222+
assert len(query_result) == TABLE_LENGTH
223223

224224
# Verify that table was split into multiple requests.
225225
assert len(futures) == 21

0 commit comments

Comments
 (0)