Skip to content

Commit 67e0f00

Browse files
committed
Better Cython types
1 parent 41a734f commit 67e0f00

File tree

2 files changed

+7
-8
lines changed

2 files changed

+7
-8
lines changed

pandas/_libs/groupby.pyx

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -408,13 +408,12 @@ def group_quantile(ndarray[float64_t] out,
408408
provided `out` parameter.
409409
"""
410410
cdef:
411-
Py_ssize_t i, N=len(labels)
412-
int64_t lab, ngroups, grp_sz, non_na_sz, grp_start=0, idx=0
413-
uint8_t interp, offset
414-
numeric val, next_val
415-
float64_t q_idx, frac
416-
ndarray[int64_t] counts, non_na_counts
417-
ndarray[int64_t] sort_arr
411+
Py_ssize_t i, N=len(labels), ngroups, grp_sz, non_na_sz
412+
Py_ssize_t grp_start=0, idx=0
413+
int64_t lab
414+
uint8_t interp
415+
float64_t q_idx, frac, val, next_val
416+
ndarray[int64_t] counts, non_na_counts, sort_arr
418417

419418
assert values.shape[0] == N
420419
inter_methods = {

pandas/tests/groupby/test_function.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1104,7 +1104,7 @@ def test_quantile(interpolation, a_vals, b_vals, q):
11041104
a_expected = pd.Series(a_vals).quantile(q, interpolation=interpolation)
11051105
b_expected = pd.Series(b_vals).quantile(q, interpolation=interpolation)
11061106

1107-
df = pd.DataFrame({
1107+
df = DataFrame({
11081108
'key': ['a'] * len(a_vals) + ['b'] * len(b_vals),
11091109
'val': a_vals + b_vals})
11101110

0 commit comments

Comments
 (0)