Skip to content

Commit 50b34a4

Browse files
authored
* TYP: nanops * cast
1 parent bdb00f2 commit 50b34a4

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

pandas/core/nanops.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -367,14 +367,21 @@ def _wrap_results(result, dtype: np.dtype, fill_value=None):
367367
return result
368368

369369

370-
def _datetimelike_compat(func):
370+
def _datetimelike_compat(func: F) -> F:
371371
"""
372372
If we have datetime64 or timedelta64 values, ensure we have a correct
373373
mask before calling the wrapped function, then cast back afterwards.
374374
"""
375375

376376
@functools.wraps(func)
377-
def new_func(values, *, axis=None, skipna=True, mask=None, **kwargs):
377+
def new_func(
378+
values: np.ndarray,
379+
*,
380+
axis: Optional[int] = None,
381+
skipna: bool = True,
382+
mask: Optional[np.ndarray] = None,
383+
**kwargs,
384+
):
378385
orig_values = values
379386

380387
datetimelike = values.dtype.kind in ["m", "M"]
@@ -390,7 +397,7 @@ def new_func(values, *, axis=None, skipna=True, mask=None, **kwargs):
390397

391398
return result
392399

393-
return new_func
400+
return cast(F, new_func)
394401

395402

396403
def _na_for_min_count(

0 commit comments

Comments
 (0)