Update total rows in progress bar during download

googleapis · Mar 25, 2019 · b23d581 · b23d581
1 parent c6efe1b
commit b23d581
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bigquery/google/cloud/bigquery/table.py b/bigquery/google/cloud/bigquery/table.py
@@ -1346,6 +1346,7 @@ def _to_dataframe_tabledata_list(self, dtypes):
             frames.append(self._to_dataframe_dtypes(page, column_names, dtypes))
 
             if pbar is not None:
+                pbar.total = pbar.total or self.total_rows
                 # update progress bar with number of rows in last frame
                 pbar.update(len(frames[-1]))