Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings
This repository was archived by the owner on May 7, 2026. It is now read-only.
42 changes: 23 additions & 19 deletions 42 bigframes/display/anywidget.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,25 +133,26 @@ def _initial_load(self) -> None:
# obtain the row counts
# TODO(b/428238610): Start iterating over the result of `to_pandas_batches()`
# before we get here so that the count might already be cached.
self._reset_batches_for_new_page_size()
with bigframes.option_context("display.progress_bar", None):
self._reset_batches_for_new_page_size()

if self._batches is None:
self._error_message = (
"Could not retrieve data batches. Data might be unavailable or "
"an error occurred."
)
self.row_count = None
elif self._batches.total_rows is None:
# Total rows is unknown, this is an expected state.
# TODO(b/461536343): Cheaply discover if we have exactly 1 page.
# There are cases where total rows is not set, but there are no additional
# pages. We could disable the "next" button in these cases.
self.row_count = None
else:
self.row_count = self._batches.total_rows
if self._batches is None:
self._error_message = (
"Could not retrieve data batches. Data might be unavailable or "
"an error occurred."
)
self.row_count = None
elif self._batches.total_rows is None:
# Total rows is unknown, this is an expected state.
# TODO(b/461536343): Cheaply discover if we have exactly 1 page.
# There are cases where total rows is not set, but there are no additional
# pages. We could disable the "next" button in these cases.
self.row_count = None
else:
self.row_count = self._batches.total_rows

# get the initial page
self._set_table_html()
# get the initial page
self._set_table_html()

@traitlets.observe("_initial_load_complete")
def _on_initial_load_complete(self, change: dict[str, Any]):
Expand Down Expand Up @@ -274,14 +275,17 @@ def _reset_batch_cache(self) -> None:

def _reset_batches_for_new_page_size(self) -> None:
"""Reset the batch iterator when page size changes."""
self._batches = self._dataframe.to_pandas_batches(page_size=self.page_size)
with bigframes.option_context("display.progress_bar", None):
self._batches = self._dataframe.to_pandas_batches(page_size=self.page_size)

self._reset_batch_cache()

def _set_table_html(self) -> None:
"""Sets the current html data based on the current page and page size."""
new_page = None
with self._setting_html_lock:
with self._setting_html_lock, bigframes.option_context(
"display.progress_bar", None
):
if self._error_message:
self.table_html = (
f"<div class='bigframes-error-message'>"
Expand Down
Loading
Morty Proxy This is a proxified and sanitized view of the page, visit original site.