Mypy/typeshed Stubs For Pandas
Solution 1:
By now (March 2021) there is a dedicated pandas stub project in progress. It is however not developed by the pandas team (thanks to @remeus for pointing that out). The data-science-types
project is deprecated. On GitHub they write:
⚠️ this project has mostly stopped development ⚠️
The pandas team and the numpy team are both in the process of integrating type stubs into their codebases, and we don't see the point of competing with them.
The NumPy
stubs used to be in a seperate project, but now have been merged into the main git repo:
These stubs have been merged into NumPy, and all further development will happen in the NumPy main repo. We welcome your contributions there!
Solution 2:
As of June 2020, pandas doesn't have stubs yet, but there is some progress in that direction. By now, you can use the data-science-types
package (on GitHub) which provides stub files for pandas
, numpy
and matplotlib
, though it's also a work in progress.
Solution 3:
I have not yet found stubs for pandas
. However, someone has created some for NumPy.
One interim option could be to define a custom type according to how mypy sees pandas objects. When I wrap a DataFrame
object in reveal_type(df)
and run mypy, it reveals that the recognized type signature is
Union[builtins.dict[Union[builtins.str, builtins.int], builtins.dict[Any, Any]], Any]
However it shows the same for a Series
object, so it's not very accurate.
Post a Comment for "Mypy/typeshed Stubs For Pandas"