format

Author: shuoweil

packages/bigframes/bigframes/display/anywidget.py

@@ -154,6 +154,7 @@ def __init__(
     @traitlets.observe("start_execution")
     def _on_start_execution(self, change: dict[str, Any]):
         if change["new"]:
+
             def run_execution():
                 try:
                     if self.is_deferred_mode:
@@ -167,24 +168,17 @@ def run_execution():
                             self._dataframe, _ = df._process_display_df()
                             self._initialize_from_dataframe()
                         elif self._dataframe is not None:
-                            self._dataframe, _ = (
-                                self._dataframe._process_display_df()
-                            )
+                            self._dataframe, _ = self._dataframe._process_display_df()
                             self._initialize_from_dataframe()
-                    elif (
-                        not self.is_deferred_mode
-                        and self._dataframe is not None
-                    ):
+                    elif not self.is_deferred_mode and self._dataframe is not None:
                         self._initial_load()
                 except Exception as e:
                     self._error_message = str(e)
                 finally:
                     self.is_deferred_mode = False
                     self.start_execution = False
 
-            self._execution_thread = threading.Thread(
-                target=run_execution, daemon=True
-            )
+            self._execution_thread = threading.Thread(target=run_execution, daemon=True)
             self._execution_thread.start()
 
     def _initialize_from_dataframe(self):
@@ -195,7 +189,6 @@ def _initialize_from_dataframe(self):
 
         self._initial_load()
 
-
     def _get_orderable_columns(
         self, dataframe: bigframes.dataframe.DataFrame
     ) -> list[str]:

packages/bigframes/tests/unit/display/test_anywidget.py

@@ -284,6 +284,7 @@ def mock_deferred_df():
         class MockDeferredBigQueryDataFrame(DeferredBigQueryDataFrame):
             def __init__(self):
                 pass
+
         df = mock.MagicMock(spec=MockDeferredBigQueryDataFrame)
         df.__class__ = DeferredBigQueryDataFrame
         yield df