Skip to content

Commit

Permalink
dask: purge unused args
Browse files Browse the repository at this point in the history
  • Loading branch information
casperdcl committed Feb 22, 2021
1 parent eb74eb6 commit 0a6f2e4
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions tqdm/dask.py
Expand Up @@ -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
Expand All @@ -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
Expand All @@ -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):
Expand Down

0 comments on commit 0a6f2e4

Please sign in to comment.