chore: move the place to take fflag

Author: shuoweil

bigframes/display/anywidget.py

@@ -275,7 +275,8 @@ def _reset_batch_cache(self) -> None:
 
     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)
+        with bigframes.option_context("display.progress_bar", None):
+            self._batches = self._dataframe.to_pandas_batches(page_size=self.page_size)
 
         self._reset_batch_cache()
 

bigframes/display/html.py

@@ -363,8 +363,7 @@ def repr_mimebundle(
 
     if opts.repr_mode == "anywidget":
         try:
-            with bigframes.option_context("display.progress_bar", None):
-                return get_anywidget_bundle(obj, include=include, exclude=exclude)
+            return get_anywidget_bundle(obj, include=include, exclude=exclude)
         except ImportError:
             # Anywidget is an optional dependency, so warn rather than fail.
             # TODO(shuowei): When Anywidget becomes the default for all repr modes,