chore: Amend astype error message and docstring (#1151)

Author: TrevorBergeron

bigframes/core/indexes/base.py

@@ -328,7 +328,7 @@ def astype(
         errors: Literal["raise", "null"] = "raise",
     ) -> Index:
         if errors not in ["raise", "null"]:
-            raise ValueError("Arg 'error' must be one of 'raise' or 'null'")
+            raise ValueError("Argument 'errors' must be one of 'raise' or 'null'")
         if self.nlevels > 1:
             raise TypeError("Multiindex does not support 'astype'")
         return self._apply_unary_expr(

bigframes/series.py

@@ -356,7 +356,7 @@ def astype(
         errors: Literal["raise", "null"] = "raise",
     ) -> Series:
         if errors not in ["raise", "null"]:
-            raise ValueError("Arg 'error' must be one of 'raise' or 'null'")
+            raise ValueError("Argument 'errors' must be one of 'raise' or 'null'")
         return self._apply_unary_op(
             bigframes.operations.AsTypeOp(to_type=dtype, safe=(errors == "null"))
         )

third_party/bigframes_vendored/pandas/core/indexes/base.py

@@ -105,8 +105,21 @@ def astype(self, dtype):
         impossible, a TypeError exception is raised.
 
         Args:
-            dtype (numpy dtype or pandas type):
-                A dtype supported by BigQuery DataFrames
+            dtype (str or pandas.ExtensionDtype):
+                A dtype supported by BigQuery DataFrame include ``'boolean'``,
+                ``'Float64'``, ``'Int64'``, ``'int64\\[pyarrow\\]'``,
+                ``'string'``, ``'string\\[pyarrow\\]'``,
+                ``'timestamp\\[us, tz=UTC\\]\\[pyarrow\\]'``,
+                ``'timestamp\\[us\\]\\[pyarrow\\]'``,
+                ``'date32\\[day\\]\\[pyarrow\\]'``,
+                ``'time64\\[us\\]\\[pyarrow\\]'``.
+                A pandas.ExtensionDtype include ``pandas.BooleanDtype()``,
+                ``pandas.Float64Dtype()``, ``pandas.Int64Dtype()``,
+                ``pandas.StringDtype(storage="pyarrow")``,
+                ``pd.ArrowDtype(pa.date32())``,
+                ``pd.ArrowDtype(pa.time64("us"))``,
+                ``pd.ArrowDtype(pa.timestamp("us"))``,
+                ``pd.ArrowDtype(pa.timestamp("us", tz="UTC"))``.
             errors ({'raise', 'null'}, default 'raise'):
                 Control raising of exceptions on invalid data for provided dtype.
                 If 'raise', allow exceptions to be raised if any value fails cast