Deduplication

Author: shuoweil

bigframes/display/anywidget.py

@@ -239,13 +239,17 @@ def _cached_data(self) -> pd.DataFrame:
             return pd.DataFrame(columns=self._dataframe.columns)
         return pd.concat(self._cached_batches, ignore_index=True)
 
+    def _reset_batch_cache(self) -> None:
+        """Resets batch caching attributes."""
+        self._cached_batches = []
+        self._batch_iter = None
+        self._all_data_loaded = False
+
     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)
 
-        self._cached_batches = []
-        self._batch_iter = None
-        self._all_data_loaded = False
+        self._reset_batch_cache()
 
     def _set_table_html(self) -> None:
         """Sets the current html data based on the current page and page size."""
@@ -265,9 +269,7 @@ def _set_table_html(self) -> None:
         # Reset batches when sorting changes
         if self._last_sort_state != _SortState(self.sort_column, self.sort_ascending):
             self._batches = df_to_display._to_pandas_batches(page_size=self.page_size)
-            self._cached_batches = []
-            self._batch_iter = None
-            self._all_data_loaded = False
+            self._reset_batch_cache()
             self._last_sort_state = _SortState(self.sort_column, self.sort_ascending)
             self.page = 0  # Reset to first page
 

notebooks/dataframes/anywidget_mode.ipynb

@@ -157,7 +157,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "dbea81204a6045a5a22ed5d368ba9095",
+       "model_id": "030f0b706a17487288677d33b6921a8e",
        "version_major": 2,
        "version_minor": 1
       },
@@ -255,7 +255,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "a62fe54c12a14609b5068d8a42e72170",
+       "model_id": "2fe252168e4249e5a694579e49f04065",
        "version_major": 2,
        "version_minor": 1
       },
@@ -369,7 +369,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "2edf57f9b10f4ebabb1925f9a0d76df3",
+       "model_id": "cc4d4dffe00a4f44b7cbcfdea440cc5a",
        "version_major": 2,
        "version_minor": 1
       },
@@ -409,7 +409,7 @@
      "data": {
       "text/html": [
        "✅ Completed. \n",
-       "    Query processed 85.9 kB in 17 seconds of slot time.\n",
+       "    Query processed 85.9 kB in 15 seconds of slot time.\n",
        "    "
       ],
       "text/plain": [
@@ -456,7 +456,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4e66779f3f0e48da9911e72c399ad960",
+       "model_id": "7f9f40cf56ec4c30a2f1f52d0e825633",
        "version_major": 2,
        "version_minor": 1
       },