File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -834,17 +834,15 @@ def iset(
834
834
835
835
# multiple columns -> convert slice or array to integer indices
836
836
elif isinstance (loc , slice ):
837
- indices = range (
837
+ indices : range | np . ndarray = range (
838
838
loc .start if loc .start is not None else 0 ,
839
839
loc .stop if loc .stop is not None else self .shape_proper [1 ],
840
840
loc .step if loc .step is not None else 1 ,
841
841
)
842
842
else :
843
843
assert isinstance (loc , np .ndarray )
844
844
assert loc .dtype == "bool"
845
- # error: Incompatible types in assignment (expression has type "ndarray",
846
- # variable has type "range")
847
- indices = np .nonzero (loc )[0 ] # type: ignore[assignment]
845
+ indices = np .nonzero (loc )[0 ]
848
846
849
847
assert value .ndim == 2
850
848
assert value .shape [0 ] == len (self ._axes [0 ])
You can’t perform that action at this time.
0 commit comments