remove anywidget from early return, allow execution proceeds to _repr_html_()

Author: shuoweil

bigframes/dataframe.py

@@ -863,6 +863,21 @@ def _repr_html_(self) -> str:
                 )
                 return formatter.repr_query_job(self._compute_dry_run())
 
+            # The anywidget frontend doesn't support the db_dtypes JSON type, so
+            # convert to strings for display.
+            json_cols = [
+                series_name
+                for series_name, series in df.items()
+                if bigframes.dtypes.contains_db_dtypes_json_dtype(series.dtype)
+            ]
+            if json_cols:
+                warnings.warn(
+                    "Converting JSON columns to strings for display. "
+                    "This is temporary and will be removed when the frontend supports JSON types."
+                )
+                for col in json_cols:
+                    df[col] = df[col]._apply_unary_op(ops.json_ops.ToJSONString())
+
             # Always create a new widget instance for each display call
             # This ensures that each cell gets its own widget and prevents
             # unintended sharing between cells

bigframes/operations/output_schemas.py

@@ -14,6 +14,8 @@
 
 import pyarrow as pa
 
+from bigframes import dtypes
+
 
 def parse_sql_type(sql: str) -> pa.DataType:
     """
@@ -43,6 +45,9 @@ def parse_sql_type(sql: str) -> pa.DataType:
     if sql.upper() == "BOOL":
         return pa.bool_()
 
+    if sql.upper() == "JSON":
+        return dtypes.JSON_ARROW_TYPE
+
     if sql.upper().startswith("ARRAY<") and sql.endswith(">"):
         inner_type = sql[len("ARRAY<") : -1]
         return pa.list_(parse_sql_type(inner_type))