From 0a6f2e4f506022731b76b99fff2e50bf5b8cac1c Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Mon, 22 Feb 2021 22:53:38 +0000 Subject: [PATCH] dask: purge unused args --- tqdm/dask.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tqdm/dask.py b/tqdm/dask.py index d4413bca5..f07d1b9ba 100644 --- a/tqdm/dask.py +++ b/tqdm/dask.py @@ -8,8 +8,7 @@ class TqdmCallback(Callback): """`dask` callback for task progress""" - def __init__(self, start=None, start_state=None, pretask=None, - posttask=None, finish=None, tqdm_class=tqdm_auto, + def __init__(self, start=None, pretask=None, tqdm_class=tqdm_auto, **tqdm_kwargs): """ Parameters @@ -19,9 +18,7 @@ def __init__(self, start=None, start_state=None, pretask=None, tqdm_kwargs : optional Any other arguments used for all bars. """ - super(TqdmCallback, self).__init__( - start=start, start_state=start_state, pretask=pretask, - posttask=posttask, finish=finish) + super(TqdmCallback, self).__init__(start=start, pretask=pretask) if tqdm_kwargs: tqdm_class = partial(tqdm_class, **tqdm_kwargs) self.tqdm_class = tqdm_class @@ -30,10 +27,10 @@ def _start_state(self, _, state): self.pbar = self.tqdm_class(total=sum( len(state[k]) for k in ['ready', 'waiting', 'running', 'finished'])) - def _posttask(self, *args, **kwargs): + def _posttask(self, *_, **__): self.pbar.update() - def _finish(self, *args, **kwargs): + def _finish(self, *_, **__): self.pbar.close() def display(self):