diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py index 735fb345363c7..54dacd2d639a6 100644 --- a/pandas/io/formats/style.py +++ b/pandas/io/formats/style.py @@ -158,13 +158,12 @@ def __init__( uuid_len: int = 5, ): # validate ordered args - if not isinstance(data, (pd.Series, pd.DataFrame)): - raise TypeError("``data`` must be a Series or DataFrame") - if data.ndim == 1: + if isinstance(data, pd.Series): data = data.to_frame() + if not isinstance(data, DataFrame): + raise TypeError("``data`` must be a Series or DataFrame") if not data.index.is_unique or not data.columns.is_unique: raise ValueError("style is not supported for non-unique indices.") - assert isinstance(data, DataFrame) self.data: DataFrame = data self.index: pd.Index = data.index self.columns: pd.Index = data.columns