remove redundant code change

Author: shuoweil

bigframes/dataframe.py

@@ -23,6 +23,7 @@
 import re
 import sys
 import textwrap
+import traceback
 import typing
 from typing import (
     Any,
@@ -856,11 +857,12 @@ def _repr_html_(self) -> str:
                 import traitlets  # noqa: F401
 
                 from bigframes import display
-            except ImportError:
+            except (AttributeError, ValueError, ImportError):
+                # Fallback if anywidget is not available
                 warnings.warn(
-                    "anywidget or its dependencies are not installed. "
+                    "Anywidget mode is not available. "
                     "Please `pip install anywidget traitlets` or `pip install 'bigframes[anywidget]'` to use interactive tables. "
-                    "Falling back to deferred mode."
+                    f"Falling back to deferred mode. Error: {traceback.format_exc()}"
                 )
                 return formatter.repr_query_job(self._compute_dry_run())
 

bigframes/display/anywidget.py

@@ -203,21 +203,19 @@ def _reset_batches_for_new_page_size(self):
 
     def _set_table_html(self):
         """Sets the current html data based on the current page and page size."""
-        # For empty dataframe, render empty table with headers.
-        if self.row_count == 0:
-            page_data = self._cached_data
-        else:
-            start = self.page * self.page_size
-            end = start + self.page_size
-
-            # fetch more data if the requested page is outside our cache
-            cached_data = self._cached_data
-            while len(cached_data) < end and not self._all_data_loaded:
-                if self._get_next_batch():
-                    cached_data = self._cached_data
-                else:
-                    break
-            page_data = cached_data.iloc[start:end]
+        start = self.page * self.page_size
+        end = start + self.page_size
+
+        # fetch more data if the requested page is outside our cache
+        cached_data = self._cached_data
+        while len(cached_data) < end and not self._all_data_loaded:
+            if self._get_next_batch():
+                cached_data = self._cached_data
+            else:
+                break
+
+        # Get the data for the current page
+        page_data = cached_data.iloc[start:end]
 
         # Generate HTML table
         self.table_html = bigframes.display.html.render_html(