Skip to content

Commit e0e0131

Browse files
committed
fix interval.repeat() typing
1 parent 8cd6b76 commit e0e0131

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pandas/core/arrays/interval.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
Sequence,
1212
Type,
1313
TypeVar,
14+
Union,
1415
cast,
1516
)
1617

@@ -1493,7 +1494,9 @@ def delete(self: IntervalArrayT, loc) -> IntervalArrayT:
14931494
return self._shallow_copy(left=new_left, right=new_right)
14941495

14951496
@Appender(_extension_array_shared_docs["repeat"] % _shared_docs_kwargs)
1496-
def repeat(self: IntervalArrayT, repeats: int, axis=None) -> IntervalArrayT:
1497+
def repeat(
1498+
self: IntervalArrayT, repeats: Union[int, Sequence[int]], axis=None
1499+
) -> IntervalArrayT:
14971500
nv.validate_repeat((), {"axis": axis})
14981501
left_repeat = self.left.repeat(repeats)
14991502
right_repeat = self.right.repeat(repeats)

0 commit comments

Comments
 (0)