cleanup type errors in tests/unit/test_pandas.py (#767)

Author: Genesis929

bigframes/core/reshape/__init__.py

@@ -116,7 +116,7 @@ def cut(
         Iterable,
     ],
     *,
-    labels: Optional[bool] = None,
+    labels: Union[Iterable[str], bool, None] = None,
 ) -> bigframes.series.Series:
     if isinstance(bins, int) and bins <= 0:
         raise ValueError("`bins` should be a positive integer.")

bigframes/pandas/__init__.py

@@ -142,7 +142,7 @@ def cut(
     x: bigframes.series.Series,
     bins: int,
     *,
-    labels: Optional[bool] = None,
+    labels: Union[Iterable[str], bool, None] = None,
 ) -> bigframes.series.Series:
     return bigframes.core.reshape.cut(
         x,

tests/unit/test_pandas.py

@@ -95,8 +95,7 @@ def test_cut_raises_with_labels():
         match="The 'labels' parameter must be either False or None.",
     ):
         mock_series = mock.create_autospec(bigframes.pandas.Series, instance=True)
-        # TODO(b/340893280): fix type error
-        bigframes.pandas.cut(mock_series, 4, labels=["a", "b", "c", "d"])  # type: ignore
+        bigframes.pandas.cut(mock_series, 4, labels=["a", "b", "c", "d"])
 
 
 @pytest.mark.parametrize(