From c319bf1de995fde62282d457e7ba10c64992235c Mon Sep 17 00:00:00 2001 From: Marcel R Date: Fri, 16 Mar 2018 17:45:43 +0100 Subject: [PATCH] Avoid hardcoded unpicklable_properties in task. --- luigi/task.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/luigi/task.py b/luigi/task.py index 890ff348e5..08e40ae214 100644 --- a/luigi/task.py +++ b/luigi/task.py @@ -34,6 +34,7 @@ import copy import functools +import luigi from luigi import six from luigi import parameter @@ -686,7 +687,7 @@ def _dump(self): pickle.dumps(self) """ - unpicklable_properties = ('set_tracking_url', 'set_status_message', 'set_progress_percentage') + unpicklable_properties = tuple(luigi.worker.TaskProcess.forward_reporter_callbacks.values()) reserved_properties = {} for property_name in unpicklable_properties: if hasattr(self, property_name):