ipython_display_ to set fallback mimetypes

Author: shuoweil

bigframes/dataframe.py

@@ -783,9 +783,7 @@ def __repr__(self) -> str:
 
         opts = bigframes.options.display
         max_results = opts.max_rows
-        # anywdiget mode uses the same display logic as the "deferred" mode
-        # for faster execution
-        if opts.repr_mode in ("deferred", "anywidget"):
+        if opts.repr_mode == "deferred":
             return formatter.repr_query_job(self._compute_dry_run())
 
         # TODO(swast): pass max_columns and get the true column count back. Maybe
@@ -831,9 +829,7 @@ def _repr_html_(self) -> str:
         """
         opts = bigframes.options.display
         max_results = opts.max_rows
-        # For anywidget mode, return deferred representation
-        # The actual widget display is handled by _ipython_display_()
-        if opts.repr_mode in ("deferred", "anywidget"):
+        if opts.repr_mode == "deferred":
             return formatter.repr_query_job(self._compute_dry_run())
 
         # Process blob columns first for non-deferred modes
@@ -910,7 +906,7 @@ def obj_ref_rt_to_html(obj_ref_rt) -> str:
         html_string += f"[{row_count} rows x {column_count} columns in total]"
         return html_string
 
-    def _ipython_display_(self):
+    def _repr_mimebundle_(self, include=None, exclude=None):
         """
         Custom display method for IPython/Jupyter environments.
         This is called by IPython's display system when the object is displayed.
@@ -938,26 +934,26 @@ def _ipython_display_(self):
 
                 # Create and display the widget
                 widget = display.TableWidget(df)
+                widget_repr = widget._repr_mimebundle_(include=include, exclude=exclude)
 
-                # IPython will automatically display the widget
-                # since we're returning it from _ipython_display_()
-                from IPython.display import display as ipython_display
-
-                ipython_display(widget)
-                return  # Important: return None to signal we handled display
+                # Use deferred repr for text/plain of anywidget display.
+                # This avoids kicking off a query when the user is just
+                # printing the last expression in a cell.
+                widget_repr["text/plain"] = repr(self)
+                widget_repr["text/html"] = self._repr_html_()
+                return widget_repr
 
             except (AttributeError, ValueError, ImportError):
                 # Fallback: let IPython use _repr_html_() instead
                 warnings.warn(
                     "Anywidget mode is not available. "
                     "Please `pip install anywidget traitlets` or `pip install 'bigframes[anywidget]'` to use interactive tables. "
-                    f"Falling back to deferred mode. Error: {traceback.format_exc()}"
+                    f"Falling back to static HTML. Error: {traceback.format_exc()}"
                 )
                 # Don't return anything - let IPython fall back to _repr_html_()
-                return
+                pass
 
-        # For other modes, don't handle display - let IPython use _repr_html_()
-        return
+        return {"text/html": self._repr_html_(), "text/plain": repr(self)}
 
     def __delitem__(self, key: str):
         df = self.drop(columns=[key])

tests/system/small/test_anywidget.py

@@ -551,21 +551,33 @@ def test_widget_row_count_reflects_actual_data_available(
         assert widget.page_size == 2  # Respects the display option
 
 
-def test_repr_html_anywidget_fallback(paginated_bf_df: bf.dataframe.DataFrame):
+def test_repr_mimebundle_anywidget_fallback(paginated_bf_df: bf.dataframe.DataFrame):
     """
-    Test that _repr_html_ falls back to deferred mode when anywidget is not available.
+    Test that _repr_mimebundle_ falls back to static html when anywidget is not available.
     """
     with bf.option_context("display.repr_mode", "anywidget"):
         # Use a mock to simulate the absence of the 'anywidget' module.
         with mock.patch.dict(
             "sys.modules", {"anywidget": None, "IPython": mock.MagicMock()}
         ):
-            # The warning is now expected inside the _ipython_display_ call, not _repr_html_
-            # The test setup doesn't easily allow capturing warnings from ipython display hooks.
-            # Instead we focus on the fallback behavior of _repr_html_
-            html = paginated_bf_df._repr_html_()
-            assert "Computation deferred." in html
-            assert "Computation will process" in html
+            bundle = paginated_bf_df._repr_mimebundle_()
+            assert "application/vnd.jupyter.widget-view+json" not in bundle
+            assert "text/html" in bundle
+            html = bundle["text/html"]
+            assert "page_1_row_1" in html
+            assert "page_1_row_2" in html
+            assert "page_2_row_1" not in html
+
+
+def test_repr_mimebundle_anywidget_success(paginated_bf_df: bf.dataframe.DataFrame):
+    """
+    Test that _repr_mimebundle_ returns a widget view when anywidget is available.
+    """
+    with bf.option_context("display.repr_mode", "anywidget"):
+        bundle = paginated_bf_df._repr_mimebundle_()
+        assert "application/vnd.jupyter.widget-view+json" in bundle
+        assert "text/html" in bundle
+        assert "text/plain" in bundle
 
 
 # TODO(b/332316283): Add tests for custom index and multiindex