From adbdfa6ffc1da610a90fe8e96c3d87ee108fa0a6 Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Tue, 26 Oct 2021 19:42:03 +0200 Subject: [PATCH] added starter/stopper to task_producer arguments, forward to _as_task --- slic/core/task/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/slic/core/task/tools.py b/slic/core/task/tools.py index 589f5b372..e4c067cda 100644 --- a/slic/core/task/tools.py +++ b/slic/core/task/tools.py @@ -2,12 +2,12 @@ from functools import wraps from slic.utils import forwards_to -def task_producer(obj, func): +def task_producer(obj, func, starter=None, stopper=None): @forwards_to(func, nfilled=1) # nfilled=1 to remove self @wraps(func) def wrapper(*args, hold=False, **kwargs): filled_func = lambda: func(*args, **kwargs) - return obj._as_task(filled_func, hold=hold) + return obj._as_task(filled_func, starter=starter, stopper=stopper, hold=hold) return wrapper