Skip to content

Commit 283147d

Browse files
committed
make sure to type check on get_loc in large hashtables
1 parent 3b01e5d commit 283147d

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

pandas/_libs/index.pyx

+2
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,8 @@ cdef class IndexEngine:
113113
if not self.is_unique:
114114
return self._get_loc_duplicates(val)
115115
values = self._get_index_values()
116+
117+
self._check_type(val)
116118
loc = _bin_search(values, val) # .searchsorted(val, side='left')
117119
if loc >= len(values):
118120
raise KeyError(val)

pandas/_libs/index_class_helper.pxi.in

+2
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@ cdef class {{name}}Engine(IndexEngine):
5151
raise KeyError(val)
5252
elif util.is_float_object(val):
5353
raise KeyError(val)
54+
elif not util.is_integer_object(val):
55+
raise KeyError(val)
5456
{{endif}}
5557

5658
{{if name != 'Object'}}

0 commit comments

Comments
 (0)