Skip to content

Commit 5c36aa1

Browse files
jbrockmendelWillAyd
authored andcommitted
TYPES: __len__, is_all_dates, inferred_type (#29475)
1 parent f4098c3 commit 5c36aa1

25 files changed

+42
-40
lines changed

pandas/_libs/hashtable_class_helper.pxi.in

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ cdef class {{name}}Vector:
100100
PyMem_Free(self.data)
101101
self.data = NULL
102102

103-
def __len__(self):
103+
def __len__(self) -> int:
104104
return self.data.n
105105

106106
cpdef to_array(self):
@@ -168,7 +168,7 @@ cdef class StringVector:
168168
PyMem_Free(self.data)
169169
self.data = NULL
170170

171-
def __len__(self):
171+
def __len__(self) -> int:
172172
return self.data.n
173173

174174
def to_array(self):
@@ -212,7 +212,7 @@ cdef class ObjectVector:
212212
self.ao = np.empty(_INIT_VEC_CAP, dtype=object)
213213
self.data = <PyObject**>self.ao.data
214214

215-
def __len__(self):
215+
def __len__(self) -> int:
216216
return self.n
217217

218218
cdef inline append(self, object obj):
@@ -270,7 +270,7 @@ cdef class {{name}}HashTable(HashTable):
270270
size_hint = min(size_hint, _SIZE_HINT_LIMIT)
271271
kh_resize_{{dtype}}(self.table, size_hint)
272272

273-
def __len__(self):
273+
def __len__(self) -> int:
274274
return self.table.size
275275

276276
def __dealloc__(self):
@@ -897,7 +897,7 @@ cdef class PyObjectHashTable(HashTable):
897897
kh_destroy_pymap(self.table)
898898
self.table = NULL
899899

900-
def __len__(self):
900+
def __len__(self) -> int:
901901
return self.table.size
902902

903903
def __contains__(self, object key):

pandas/_libs/internals.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ cdef class BlockPlacement:
6666
def __repr__(self) -> str:
6767
return str(self)
6868

69-
def __len__(self):
69+
def __len__(self) -> int:
7070
cdef:
7171
slice s = self._ensure_has_slice()
7272
if s is not None:

pandas/core/arrays/categorical.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1940,7 +1940,7 @@ def take_nd(self, indexer, allow_fill=None, fill_value=None):
19401940

19411941
take = take_nd
19421942

1943-
def __len__(self):
1943+
def __len__(self) -> int:
19441944
"""
19451945
The length of this Categorical.
19461946
"""

pandas/core/arrays/datetimelike.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,7 @@ def size(self) -> int:
396396
"""The number of elements in this array."""
397397
return np.prod(self.shape)
398398

399-
def __len__(self):
399+
def __len__(self) -> int:
400400
return len(self._data)
401401

402402
def __getitem__(self, key):

pandas/core/arrays/integer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ def __setitem__(self, key, value):
469469
self._data[key] = value
470470
self._mask[key] = mask
471471

472-
def __len__(self):
472+
def __len__(self) -> int:
473473
return len(self._data)
474474

475475
@property

pandas/core/arrays/interval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,7 @@ def _validate(self):
489489
def __iter__(self):
490490
return iter(np.asarray(self))
491491

492-
def __len__(self):
492+
def __len__(self) -> int:
493493
return len(self.left)
494494

495495
def __getitem__(self, value):

pandas/core/computation/expr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -837,7 +837,7 @@ def __call__(self):
837837
def __repr__(self) -> str:
838838
return printing.pprint_thing(self.terms)
839839

840-
def __len__(self):
840+
def __len__(self) -> int:
841841
return len(self.expr)
842842

843843
def parse(self):

pandas/core/frame.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1023,7 +1023,7 @@ def itertuples(self, index=True, name="Pandas"):
10231023
# fallback to regular tuples
10241024
return zip(*arrays)
10251025

1026-
def __len__(self):
1026+
def __len__(self) -> int:
10271027
"""
10281028
Returns length of info axis, but here we use the index.
10291029
"""

pandas/core/generic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1952,7 +1952,7 @@ def items(self):
19521952
def iteritems(self):
19531953
return self.items()
19541954

1955-
def __len__(self):
1955+
def __len__(self) -> int:
19561956
"""Returns length of info axis"""
19571957
return len(self._info_axis)
19581958

pandas/core/groupby/groupby.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,7 @@ def __init__(
399399
# we accept no other args
400400
validate_kwargs("group", kwargs, {})
401401

402-
def __len__(self):
402+
def __len__(self) -> int:
403403
return len(self.groups)
404404

405405
def __repr__(self) -> str:

0 commit comments

Comments
 (0)