final touch up

Author: shuoweil

bigframes/display/anywidget.py

@@ -74,7 +74,7 @@ def __init__(self, dataframe: bigframes.dataframe.DataFrame):
         # Initialize data fetching attributes.
         self._batches = dataframe.to_pandas_batches(page_size=initial_page_size)
 
-        # Now it's safe to set traitlets properties that trigger observers
+        # set traitlets properties that trigger observers
         self.page_size = initial_page_size
 
         # len(dataframe) is expensive, since it will trigger a
@@ -102,12 +102,15 @@ def _css(self):
     table_html = traitlets.Unicode().tag(sync=True)
 
     @traitlets.validate("page")
-    def _validate_page(self, proposal: Dict[str, Any]):
+    def _validate_page(self, proposal: Dict[str, Any]) -> int:
         """Validate and clamp the page number to a valid range.
 
         Args:
             proposal: A dictionary from the traitlets library containing the
                 proposed change. The new value is in proposal["value"].
+
+        Returns:
+            The validated and clamped page number as an integer.
         """
 
         value = proposal["value"]
@@ -121,11 +124,15 @@ def _validate_page(self, proposal: Dict[str, Any]):
         return max(0, min(value, max_page))
 
     @traitlets.validate("page_size")
-    def _validate_page_size(self, proposal: Dict[str, Any]):
+    def _validate_page_size(self, proposal: Dict[str, Any]) -> int:
         """Validate page size to ensure it's positive and reasonable.
+
         Args:
             proposal: A dictionary from the traitlets library containing the
                 proposed change. The new value is in proposal["value"].
+
+        Returns:
+            The validated page size as an integer.
         """
         value = proposal["value"]
 
@@ -141,7 +148,7 @@ def _get_next_batch(self) -> bool:
         """
         Gets the next batch of data from the generator and appends to cache.
 
-        Return:
+        Returns:
             True if a batch was successfully loaded, False otherwise.
         """
         if self._all_data_loaded:

notebooks/dataframes/anywidget_mode.ipynb

@@ -75,7 +75,7 @@
     {
      "data": {
       "text/html": [
-       "Query job 0b22b0f5-b952-4546-a969-41a89e343e9b is DONE. 0 Bytes processed. <a target=\"_blank\" href=\"https://console.cloud.google.com/bigquery?project=bigframes-dev&j=bq:US:0b22b0f5-b952-4546-a969-41a89e343e9b&page=queryresults\">Open Job</a>"
+       "Query job c5fcfd5e-1617-49c8-afa3-86ca21019de4 is DONE. 0 Bytes processed. <a target=\"_blank\" href=\"https://console.cloud.google.com/bigquery?project=bigframes-dev&j=bq:US:c5fcfd5e-1617-49c8-afa3-86ca21019de4&page=queryresults\">Open Job</a>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -141,7 +141,7 @@
     {
      "data": {
       "text/html": [
-       "Query job 8e57da45-b6a7-44fb-8c4f-4b87058d94cb is DONE. 171.4 MB processed. <a target=\"_blank\" href=\"https://console.cloud.google.com/bigquery?project=bigframes-dev&j=bq:US:8e57da45-b6a7-44fb-8c4f-4b87058d94cb&page=queryresults\">Open Job</a>"
+       "Query job ab900a53-5011-4e80-85d5-0ef2958598db is DONE. 171.4 MB processed. <a target=\"_blank\" href=\"https://console.cloud.google.com/bigquery?project=bigframes-dev&j=bq:US:ab900a53-5011-4e80-85d5-0ef2958598db&page=queryresults\">Open Job</a>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -153,7 +153,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4d00aaf284984cbc97483c651b9c5110",
+       "model_id": "bda63ba739dc4d5f83a5e18eb27b2686",
        "version_major": 2,
        "version_minor": 1
       },
@@ -204,7 +204,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "d4af4cf7d24d4f1c8e9c9b5f237df32b",
+       "model_id": "9bffeb73549e48419c3dd895edfe30e8",
        "version_major": 2,
        "version_minor": 1
       },
@@ -290,7 +290,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "0f04ad3c464145ee9735eba09f5107a9",
+       "model_id": "dfd4fa3a1c6f4b3eb1877cb0e9ba7e94",
        "version_major": 2,
        "version_minor": 1
       },