From 8b37bacee5669e4b28122be0323d13972cf95500 Mon Sep 17 00:00:00 2001 From: Maxim Ivanov Date: Thu, 10 Sep 2020 13:02:15 +0700 Subject: [PATCH] TYP: alias IndexLabel without Optional --- pandas/_typing.py | 2 +- pandas/core/generic.py | 2 +- pandas/io/formats/csvs.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pandas/_typing.py b/pandas/_typing.py index 7aef5c02e290f..16d81c0d39cbe 100644 --- a/pandas/_typing.py +++ b/pandas/_typing.py @@ -83,7 +83,7 @@ Axis = Union[str, int] Label = Optional[Hashable] -IndexLabel = Optional[Union[Label, Sequence[Label]]] +IndexLabel = Union[Label, Sequence[Label]] Level = Union[Label, int] Ordered = Optional[bool] JSONSerializable = Optional[Union[PythonScalar, List, Dict]] diff --git a/pandas/core/generic.py b/pandas/core/generic.py index fffd2e068ebcf..6dbf6866fd129 100644 --- a/pandas/core/generic.py +++ b/pandas/core/generic.py @@ -3161,7 +3161,7 @@ def to_csv( columns: Optional[Sequence[Label]] = None, header: Union[bool_t, List[str]] = True, index: bool_t = True, - index_label: IndexLabel = None, + index_label: Optional[IndexLabel] = None, mode: str = "w", encoding: Optional[str] = None, compression: CompressionOptions = "infer", diff --git a/pandas/io/formats/csvs.py b/pandas/io/formats/csvs.py index 90ab6f61f4d74..bece82d840e99 100644 --- a/pandas/io/formats/csvs.py +++ b/pandas/io/formats/csvs.py @@ -42,7 +42,7 @@ def __init__( cols: Optional[Sequence[Label]] = None, header: Union[bool, Sequence[Hashable]] = True, index: bool = True, - index_label: IndexLabel = None, + index_label: Optional[IndexLabel] = None, mode: str = "w", encoding: Optional[str] = None, errors: str = "strict", @@ -100,7 +100,7 @@ def index_label(self) -> IndexLabel: return self._index_label @index_label.setter - def index_label(self, index_label: IndexLabel) -> None: + def index_label(self, index_label: Optional[IndexLabel]) -> None: if index_label is not False: if index_label is None: index_label = self._get_index_label_from_obj()