fix: suppress JSONDtypeWarning in Anywidget mode and clean up progress output (#2441)

This PR improves the user experience when using the interactive
anywidget display mode (bpd.options.display.repr_mode = "anywidget") by
reducing console noise.
Verified at:
vs code notebook: screen/ACCJRLwyThciMk8
colab notebook: screen/BhNxzpvckYg9Wp8
Fixes #<482120359> 🦕

Author: shuoweil

bigframes/__init__.py

@@ -14,13 +14,26 @@
 
 """BigQuery DataFrames provides a DataFrame API scaled by the BigQuery engine."""
 
-frombigframes._configimportoption_context, options
-frombigframes._config.bigquery_optionsimportBigQueryOptions
-frombigframes.core.global_sessionimportclose_session, get_global_session
-importbigframes.enumsasenums
-importbigframes.exceptionsasexceptions
-frombigframes.sessionimportconnect, Session
-frombigframes.versionimport__version__
+importwarnings
+
+# Suppress Python version support warnings from google-cloud libraries.
+# These are particularly noisy in Colab which still uses Python 3.10.
+warnings.filterwarnings(
+"ignore",
+category=FutureWarning,
+message=".*Google will stop supporting.*Python.*",
+)
+
+frombigframes._configimportoption_context, options# noqa: E402
+frombigframes._config.bigquery_optionsimportBigQueryOptions# noqa: E402
+frombigframes.core.global_sessionimport ( # noqa: E402
+close_session,
+get_global_session,
+)
+importbigframes.enumsasenums# noqa: E402
+importbigframes.exceptionsasexceptions# noqa: E402
+frombigframes.sessionimportconnect, Session# noqa: E402
+frombigframes.versionimport__version__# noqa: E402
 
 _MAGIC_NAMES= ["bqsql"]
 

bigframes/dataframe.py

@@ -332,7 +332,7 @@ def dtypes(self) -> pandas.Series:
 
 @property
 defcolumns(self) ->pandas.Index:
-returnself.dtypes.index
+returnself._block.column_labels
 
 @columns.setter
 defcolumns(self, labels: pandas.Index):

bigframes/display/anywidget.py

@@ -23,6 +23,7 @@
 importthreading
 fromtypingimportAny, Iterator, Optional
 importuuid
+importwarnings
 
 importpandasaspd
 
@@ -111,23 +112,32 @@ def __init__(self, dataframe: bigframes.dataframe.DataFrame):
 self.page_size=initial_page_size
 self.max_columns=initial_max_columns
 
-# TODO(b/469861913): Nested columns from structs (e.g., 'struct_col.name') are not currently sortable.
-# TODO(b/463754889): Support non-string column labels for sorting.
-ifall(isinstance(col, str) forcolindataframe.columns):
-self.orderable_columns= [
-str(col_name)
-forcol_name, dtypeindataframe.dtypes.items()
-ifdtypes.is_orderable(dtype)
- ]
-else:
-self.orderable_columns= []
+self.orderable_columns=self._get_orderable_columns(dataframe)
 
 self._initial_load()
 
 # 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
 
+def_get_orderable_columns(
+self, dataframe: bigframes.dataframe.DataFrame
+ ) ->list[str]:
+"""Determine which columns can be used for client-side sorting."""
+# TODO(b/469861913): Nested columns from structs (e.g., 'struct_col.name') are not currently sortable.
+# TODO(b/463754889): Support non-string column labels for sorting.
+ifnotall(isinstance(col, str) forcolindataframe.columns):
+return []
+
+withwarnings.catch_warnings():
+warnings.simplefilter("ignore", bigframes.exceptions.JSONDtypeWarning)
+warnings.simplefilter("ignore", category=FutureWarning)
+return [
+str(col_name)
+forcol_name, dtypeindataframe.dtypes.items()
+ifdtypes.is_orderable(dtype)
+ ]
+
 def_initial_load(self) ->None:
 """Get initial data and row count."""
 # obtain the row counts

bigframes/display/html.py

@@ -363,7 +363,13 @@ def repr_mimebundle(
 
 ifopts.repr_mode=="anywidget":
 try:
-returnget_anywidget_bundle(obj, include=include, exclude=exclude)
+withbigframes.option_context("display.progress_bar", None):
+withwarnings.catch_warnings():
+warnings.simplefilter(
+"ignore", category=bigframes.exceptions.JSONDtypeWarning
+ )
+warnings.simplefilter("ignore", category=FutureWarning)
+returnget_anywidget_bundle(obj, include=include, exclude=exclude)
 exceptImportError:
 # Anywidget is an optional dependency, so warn rather than fail.
 # TODO(shuowei): When Anywidget becomes the default for all repr modes,

bigframes/pandas/io/api.py

@@ -354,15 +354,14 @@ def _read_gbq_colab(
  )
 _set_default_session_location_if_possible_deferred_query(create_query)
 ifnotconfig.options.bigquery._session_started:
-withwarnings.catch_warnings():
-# Don't warning about Polars in SQL cell.
-# Related to b/437090788.
-try:
-bigframes._importing.import_polars()
-warnings.simplefilter("ignore", bigframes.exceptions.PreviewWarning)
-config.options.bigquery.enable_polars_execution=True
-exceptImportError:
-pass# don't fail if polars isn't available
+# Don't warning about Polars in SQL cell.
+# Related to b/437090788.
+try:
+bigframes._importing.import_polars()
+warnings.simplefilter("ignore", bigframes.exceptions.PreviewWarning)
+config.options.bigquery.enable_polars_execution=True
+exceptImportError:
+pass# don't fail if polars isn't available
 
 returnglobal_session.with_default_session(
 bigframes.session.Session._read_gbq_colab,