Missing total for DataFrame.progress_apply

[kpj]

Hello,

I am using tqdm version 4.11.2 and pandas version 0.19.2.
When I try the minimal example

import pandas as pd
from tqdm import tqdm

tqdm.pandas()
df = pd.DataFrame({1:[1], 2:[2], 3:[3]})
foo = df.progress_apply(lambda x: x+1)

the output is

4it [00:00, 2616.13it/s]

instead of a progressbar counting up to 4.

Am I missing something? I took a short peek at the code, but total seems to be set correctly, so I am not sure what exactly is going on.

#299 seems to be a similar issue, but focuses on a different problem, I think.

[cancan101]

For some reason the total is set for Series but not for DataFrames. See:

tqdm/tqdm/_tqdm.py

Lines 511 to 512 in ed6e0b4

	total = len(df) if isinstance(df, Series) \
	else df.size // len(df)

:

total = len(df) if isinstance(df, Series) else df.size // len(df)

where: df.size // len(df)==1

[cancan101]

I think the issue is whether the apply is along axis 0 or axis 1.

[kpj]

Perfect, it seems to work now.
Thanks a lot!

[casperdcl]

Great. Note to self: we should probably add a unit test to prevent against future regressions

[casperdcl]

think unit tests are now in #524