diff --git a/pandas/core/groupby/groupby.py b/pandas/core/groupby/groupby.py index a9ce8e11ddcdb..6c105b6a51e3e 100644 --- a/pandas/core/groupby/groupby.py +++ b/pandas/core/groupby/groupby.py @@ -3303,13 +3303,9 @@ def post_processor( orig_scalar = is_scalar(q) if orig_scalar: - # error: Incompatible types in assignment (expression has type "List[ - # Union[float, ExtensionArray, ndarray[Any, Any], Index, Series]]", - # variable has type "Union[float, Union[Union[ExtensionArray, ndarray[ - # Any, Any]], Index, Series]]") - q = [q] # type: ignore[assignment] - - qs = np.array(q, dtype=np.float64) + qs = np.array([q], dtype=np.float64) + else: + qs = np.array(q, dtype=np.float64) ids, _, ngroups = self.grouper.group_info nqs = len(qs)