fix mypy

Author: shuoweil

bigframes/display/anywidget.py

@@ -136,13 +136,17 @@ def __init__(self, dataframe: bigframes.dataframe.DataFrame):
             self.row_count = self._batches.total_rows
 
         # get the initial page
-        self._get_next_batch()
         self._set_table_html()
 
         # Signals to the frontend that the initial data load is complete.
         # Also used as a guard to prevent observers from firing during initialization.
         self._initial_load_complete = True
 
+    @traitlets.observe("_initial_load_complete")
+    def _on_initial_load_complete(self, change: Dict[str, Any]):
+        if change["new"]:
+            self._set_table_html()
+
     @functools.cached_property
     def _esm(self):
         """Load JavaScript code from external file."""

notebooks/dataframes/anywidget_mode.ipynb

@@ -157,7 +157,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "3ad3ce907f4d43de8c3c15f0d1cdebd6",
+       "model_id": "2935e3f8f4f34c558d588f09a9c42131",
        "version_major": 2,
        "version_minor": 1
       },
@@ -255,7 +255,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "684becdc45de4a15826672c0a7e89930",
+       "model_id": "fa03d998dfee47638a32b6c21ace0b5c",
        "version_major": 2,
        "version_minor": 1
       },
@@ -369,7 +369,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "5cccf9eb2a674f9f8a1bdb165921d5f9",
+       "model_id": "6d6886bd2bb74be996d54ce240cbe6c9",
        "version_major": 2,
        "version_minor": 1
       },
@@ -401,15 +401,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 11,
    "id": "added-cell-1",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/html": [
        "✅ Completed. \n",
-       "    Query processed 85.9 kB in 13 seconds of slot time.\n",
+       "    Query processed 85.9 kB in 24 seconds of slot time.\n",
        "    "
       ],
       "text/plain": [
@@ -456,7 +456,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "06e704906f254de194e5e3ea1ef2bdbb",
+       "model_id": "df774329fd2f47918b986362863d7155",
        "version_major": 2,
        "version_minor": 1
       },
@@ -474,7 +474,7 @@
        "Computation deferred. Computation will process 0 Bytes"
       ]
      },
-     "execution_count": 10,
+     "execution_count": 11,
      "metadata": {},
      "output_type": "execute_result"
     }