Skip to content

Commit d2445df

Browse files
committed
fix errors
1 parent 81b014f commit d2445df

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

pandas/core/indexes/category.py

+1
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,7 @@ def astype(self, dtype, copy: bool = True) -> Index:
286286

287287
cat = self.categories
288288
if type(cat) is NumIndex:
289+
assert isinstance(cat, NumIndex) # mypy complaint fix
289290
try:
290291
cat._validate_dtype(dtype)
291292
except ValueError:

pandas/tests/base/test_unique.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_unique(index_or_series_obj):
2424
expected = pd.MultiIndex.from_tuples(unique_values)
2525
expected.names = obj.names
2626
tm.assert_index_equal(result, expected)
27-
elif isinstance(obj, pd.NumIndex):
27+
elif type(obj) is pd.NumIndex:
2828
expected = pd.NumIndex(unique_values, dtype=obj.dtype)
2929
tm.assert_index_equal(result, expected)
3030
elif isinstance(obj, pd.Index):
@@ -65,7 +65,7 @@ def test_unique_null(null_obj, index_or_series_obj):
6565
unique_values_not_null = [val for val in unique_values_raw if not pd.isnull(val)]
6666
unique_values = [null_obj] + unique_values_not_null
6767

68-
if isinstance(obj, pd.NumIndex):
68+
if type(obj) is pd.NumIndex:
6969
expected = pd.NumIndex(unique_values, dtype=obj.dtype)
7070
tm.assert_index_equal(result, expected)
7171
elif isinstance(obj, pd.Index):

0 commit comments

Comments
 (0)