Problem with progress apply

[acampove]

Hi,

The code below:

import pandas as pd
from tqdm import tqdm

# Enable tqdm for pandas
tqdm.pandas()

# Create a dummy DataFrame
df = pd.DataFrame({'x': [1, 2, 3]})

# Define a simple row function
def double(row: pd.Series) -> pd.Series:
    row['x'] = row['x'] * 2
    return row

# Apply with progress_apply
df_filled = df.progress_apply(double, axis=1)
print(df_filled)

shows me:

I am using pyright 1.1.404 and:

(rx) [acampove@thinkbook types]$ pip show pandas-stubs
Name: pandas-stubs
Version: 2.3.0.250703
Summary: Type annotations for pandas
Home-page: https://pandas.pydata.org
Author: The Pandas Development Team
Author-email: pandas-dev@python.org
License: BSD-3-Clause
Location: /home/acampove/micromamba/envs/rx/lib/python3.12/site-packages
Requires: numpy, types-pytz
Required-by:

Please complete the following information:

• OS: almalinux9
• OS Version: 5.14.0-570.24.1.el9_6.x86_64 Tasks to complete before making this repo public #1 SMP PREEMPT_DYNAMIC Tue Jul 1 04:35:16 EDT 2025 x86_64 x86_64 x86_64 GNU/Linux
• python version 3.12
• version of type checker 1.1.404
• version of installed pandas-stubs see above

[Dr-Irv]

I don't think we can fix this. from the pandas perspective, progress_apply is just like a column in a DataFrame, so it is typed as a Series.

What tqdm is doing is extending pandas in a dynamic way, but you are using a static type checker, and progress_apply is not defined in a static way.