function call updated in

  bigframes/core/blocks.py, unused function removed from bigframes/dtypes.py

Author: shuoweil

bigframes/core/blocks.py

@@ -721,9 +721,7 @@ def to_pandas_batches(
                 # db_dtypes.JSONArrowType, especially when nested.
                 # Create with string type and then cast.
                 if isinstance(dtype, pd.ArrowDtype):
-                    safe_pa_type = bigframes.dtypes._replace_json_arrow_with_string(
-                        dtype.pyarrow_dtype
-                    )
+                    safe_pa_type = bigframes.dtypes.to_storage_type(dtype.pyarrow_dtype)
                     # Create empty array with safe type, but preserve original dtype metadata
                     empty_array = pa.array([], type=safe_pa_type)
                     series_map[col] = pd.Series(

bigframes/dtypes.py

@@ -972,21 +972,6 @@ def contains_db_dtypes_json_dtype(dtype):
     return contains_db_dtypes_json_arrow_type(dtype.pyarrow_dtype)
 
 
-def _replace_json_arrow_with_string(pa_type: pa.DataType) -> pa.DataType:
-    """Recursively replace JSONArrowType with string type."""
-    if isinstance(pa_type, db_dtypes.JSONArrowType):
-        return pa.string()
-    if isinstance(pa_type, pa.ListType):
-        return pa.list_(_replace_json_arrow_with_string(pa_type.value_type))
-    if isinstance(pa_type, pa.StructType):
-        new_fields = [
-            field.with_type(_replace_json_arrow_with_string(field.type))
-            for field in pa_type
-        ]
-        return pa.struct(new_fields)
-    return pa_type
-
-
 def warn_on_db_dtypes_json_dtype(dtypes):
     """Warn that the JSON dtype is changing.