Closed
Description
Describe the bug
Method .stack
of dataframe is not fully typed, it misses argument sort
and future_stack
To Reproduce
- Provide a minimal runnable
pandas
example that is not properly checked by the stubs.
import pandas as pd
df = pd.DataFrame([1, 2, 3])
df.stack(future_stack=True)
df.stack(sort=True)
- Indicate which type checker you are using (
mypy
orpyright
).
I am using mypy (1.10.0) - Show the error message received from that type checker while checking your example.
ex.py:3: error: Unexpected keyword argument "future_stack" for "stack" of "DataFrame" [call-arg]
ex.py:4: error: Unexpected keyword argument "sort" for "stack" of "DataFrame" [call-arg]
Found 2 errors in 1 file (checked 1 source file)
Please complete the following information:
- OS: ArchLinux
- python version: 3.10.14
- version of type checker: 1.10.0
- version of installed
pandas-stubs
2.2.2.240514
Additional context
I'll be glad to open a PR to add those two arguments with some tests if that's ok for you, but I'm not entirely sure about what is the philosophy about pandas-stubs here. Since they are new arguments in 2.1, could it be bad to add those arguments in pandas stubs for someone who sill uses 2.0?