Skip to content

Commit 76ffca9

Browse files
GH804 PR feedback
1 parent 0caec38 commit 76ffca9

File tree

2 files changed

+12
-9
lines changed

2 files changed

+12
-9
lines changed

pandas-stubs/core/indexes/base.pyi

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -315,13 +315,13 @@ class Index(IndexOpsMixin[S1]):
315315
def to_series(self, index=..., name: Hashable = ...) -> Series: ...
316316
def to_frame(self, index: bool = True, name=...) -> DataFrame: ...
317317
@property
318-
def name(self) -> _str | None: ...
318+
def name(self) -> Hashable | None: ...
319319
@name.setter
320-
def name(self, value) -> None: ...
320+
def name(self, value: Hashable) -> None: ...
321321
@property
322-
def names(self) -> list[_str | None]: ...
322+
def names(self) -> list[Hashable | None]: ...
323323
@names.setter
324-
def names(self, names: SequenceNotStr[_str | None]) -> None: ...
324+
def names(self, names: SequenceNotStr[Hashable | None]) -> None: ...
325325
def set_names(self, names, *, level=..., inplace: bool = ...): ...
326326
@overload
327327
def rename(self, name, *, inplace: Literal[False] = False) -> Self: ...

tests/test_indexes.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from __future__ import annotations
22

3+
from collections.abc import Hashable
34
import datetime as dt
45
from typing import (
56
TYPE_CHECKING,
@@ -1439,15 +1440,17 @@ def test_index_naming() -> None:
14391440
"""
14401441
Test index names type both for the getter and the setter.
14411442
The names of an index should be settable with a sequence (not str) and names
1442-
property is a list[str | None] (FrozenList).
1443+
property is a list[Hashable | None] (FrozenList).
14431444
"""
14441445
df = pd.DataFrame({"a": ["a", "b", "c"], "i": [10, 11, 12]})
14451446

14461447
df.index.names = ["idx"]
1447-
check(assert_type(df.index.names, list[str | None]), list)
1448+
check(assert_type(df.index.names, list[Hashable | None]), list)
1449+
df.index.names = [3]
1450+
check(assert_type(df.index.names, list[Hashable | None]), list)
14481451
df.index.names = ("idx2",)
1449-
check(assert_type(df.index.names, list[str | None]), list)
1452+
check(assert_type(df.index.names, list[Hashable | None]), list)
14501453
df.index.names = [None]
1451-
check(assert_type(df.index.names, list[str | None]), list)
1454+
check(assert_type(df.index.names, list[Hashable | None]), list)
14521455
df.index.names = (None,)
1453-
check(assert_type(df.index.names, list[str | None]), list)
1456+
check(assert_type(df.index.names, list[Hashable | None]), list)

0 commit comments

Comments
 (0)