¿Cómo usar tqdm con pandas en un cuaderno jupyter?

Estoy haciendo un análisis con pandas en un cuaderno jupyter y, dado que mi función de aplicación lleva mucho tiempo, me gustaría ver una barra de progreso. A través de esta publicaciónaquí Encontré la biblioteca tqdm que proporciona una barra de progreso simple paraoperaciones de pandas. También hay unaIntegración Jupyter eso proporciona una barra de progreso realmente agradable donde la barra en sí misma cambia con el tiempo.

Sin embargo, me gustaría combinar los dos y no entiendo cómo hacerlo. Tomemos el mismo ejemplo que en la documentación.

import pandas as pd
import numpy as np
from tqdm import tqdm

df = pd.DataFrame(np.random.randint(0, 100, (100000, 6)))

# Register `pandas.progress_apply` and `pandas.Series.map_apply` with `tqdm`
# (can use `tqdm_gui`, `tqdm_notebook`, optional kwargs, etc.)
tqdm.pandas(desc="my bar!")

# Now you can use `progress_apply` instead of `apply`
# and `progress_map` instead of `map`
df.progress_apply(lambda x: x**2)
# can also groupby:
# df.groupby(0).progress_apply(lambda x: x**2)

Incluso dice "puede usar 'tqdm_notebook'" pero no encuentro la forma de hacerlo. He intentado algunas cosas como

tqdm_notebook(tqdm.pandas(desc="my bar!"))

o

tqdm_notebook.pandas

Pero no funcionan. En eldefinición me parece como

tqdm.pandas(tqdm_notebook(desc="my bar!"))

debería funcionar, pero la barra no muestra correctamente el progreso y todavía hay resultados adicionales.

¿Alguna otra idea?