Commit3e06961

authored

Close the to_dataframe progress bar when finished. (#7757)

1 parentee804a1 commit3e06961Copy full SHA for 3e06961

File tree

-0

lines changed

-0

lines changed

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -1385,6 +1385,10 @@ def _to_dataframe_tabledata_list(self, dtypes, progress_bar=None):`
`1385`	`1385`	`progress_bar.total=progress_bar.totalorself.total_rows`
`1386`	`1386`	`progress_bar.update(len(current_frame))`
`1387`	`1387`
	`1388`	`+ifprogress_barisnotNone:`
	`1389`	`+# Indicate that the download has finished.`
	`1390`	`+progress_bar.close()`
	`1391`	`+`
`1388`	`1392`	`returnpandas.concat(frames)`
`1389`	`1393`
`1390`	`1394`	`def_to_dataframe_bqstorage_stream(`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -1503,6 +1503,7 @@ def test_to_dataframe_progress_bar(`
`1503`	`1503`
`1504`	`1504`	`progress_bar_mock.assert_called()`
`1505`	`1505`	`progress_bar_mock().update.assert_called()`
	`1506`	`+progress_bar_mock().close.assert_called_once()`
`1506`	`1507`	`self.assertEqual(len(df),4)`
`1507`	`1508`
`1508`	`1509`	@unittest.skipIf(pandasisNone,"Requires `pandas`")

Comments

(0)