25 lines
888 B
Python
25 lines
888 B
Python
import sys
|
|
|
|
__author__ = "github.com/casperdcl"
|
|
__all__ = ['tqdm_pandas']
|
|
|
|
|
|
def tqdm_pandas(tclass, **tqdm_kwargs):
|
|
"""
|
|
Registers the given `tqdm` instance with
|
|
`pandas.core.groupby.DataFrameGroupBy.progress_apply`.
|
|
"""
|
|
from tqdm import TqdmDeprecationWarning
|
|
|
|
if isinstance(tclass, type) or (getattr(tclass, '__name__', '').startswith(
|
|
'tqdm_')): # delayed adapter case
|
|
TqdmDeprecationWarning(
|
|
"Please use `tqdm.pandas(...)` instead of `tqdm_pandas(tqdm, ...)`.",
|
|
fp_write=getattr(tqdm_kwargs.get('file', None), 'write', sys.stderr.write))
|
|
tclass.pandas(**tqdm_kwargs)
|
|
else:
|
|
TqdmDeprecationWarning(
|
|
"Please use `tqdm.pandas(...)` instead of `tqdm_pandas(tqdm(...))`.",
|
|
fp_write=getattr(tclass.fp, 'write', sys.stderr.write))
|
|
type(tclass).pandas(deprecated_t=tclass)
|