search

wearepal / data-science-types

Mypy stubs, i.e., type information, for numpy, pandas and matplotlib
Apache License 2.0
202 stars 51 forks source link

Fails type checking on dataframe.to_pickle #216

Closed EdwardJRoss closed 3 years ago

EdwardJRoss commented 3 years ago

Running mypy with data-science-types on the following

import pandas as pd
df = pd.DataFrame({'a': [1]})
df.to_pickle('output.pkl')

Produces an error:

error: "Series[Any]" not callable

I would expect it to pass using DataFrame.to_pickle