Skip to content

Commit 0fc8740

Browse files
committed
fixes
1 parent a5e661a commit 0fc8740

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

pandas/core/indexes/range.py

+11-10
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def _get_indexer(
428428
def repeat(self, repeats, axis=None) -> Int64Index:
429429
return self._int64index.repeat(repeats, axis=axis)
430430

431-
def delete(self, loc) -> Int64Index:
431+
def delete(self, loc) -> Int64Index: # type: ignore[override]
432432
return self._int64index.delete(loc)
433433

434434
def take(
@@ -749,7 +749,7 @@ def symmetric_difference(self, other, result_name: Hashable = None, sort=None):
749749

750750
# --------------------------------------------------------------------
751751

752-
def _concat(self, indexes: list[Index], name: Hashable):
752+
def _concat(self, indexes: list[Index], name: Hashable) -> Index:
753753
"""
754754
Overriding parent method for the case of all RangeIndex instances.
755755
@@ -761,16 +761,15 @@ def _concat(self, indexes: list[Index], name: Hashable):
761761
if not all(isinstance(x, RangeIndex) for x in indexes):
762762
return super()._concat(indexes, name)
763763

764-
elif len(indexes) == 1:
765-
return indexes[0]
764+
rng_indexes: list[RangeIndex] = [cast(RangeIndex, obj) for obj in indexes]
765+
766+
if len(indexes) == 1:
767+
return rng_indexes[0]
766768

767769
start = step = next_ = None
768770

769771
# Filter the empty indexes
770-
assert all(isinstance(x, RangeIndex) for x in indexes)
771-
non_empty_indexes: list[RangeIndex] = [
772-
cast(RangeIndex, obj) for obj in indexes if len(obj)
773-
]
772+
non_empty_indexes = [obj for obj in rng_indexes if len(obj)]
774773

775774
for obj in non_empty_indexes:
776775
rng = obj._range
@@ -783,7 +782,9 @@ def _concat(self, indexes: list[Index], name: Hashable):
783782
elif step is None:
784783
# First non-empty index had only one element
785784
if rng.start == start:
786-
result = Int64Index(np.concatenate([x._values for x in indexes]))
785+
result = Int64Index(
786+
np.concatenate([x._values for x in rng_indexes])
787+
)
787788
return result.rename(name)
788789

789790
step = rng.start - start
@@ -792,7 +793,7 @@ def _concat(self, indexes: list[Index], name: Hashable):
792793
next_ is not None and rng.start != next_
793794
)
794795
if non_consecutive:
795-
result = Int64Index(np.concatenate([x._values for x in indexes]))
796+
result = Int64Index(np.concatenate([x._values for x in rng_indexes]))
796797
return result.rename(name)
797798

798799
if step is not None:

0 commit comments

Comments
 (0)