diff --git a/datajob/stepfunctions/stepfunctions_workflow.py b/datajob/stepfunctions/stepfunctions_workflow.py index ba038d4..487cc96 100644 --- a/datajob/stepfunctions/stepfunctions_workflow.py +++ b/datajob/stepfunctions/stepfunctions_workflow.py @@ -63,6 +63,7 @@ def __init__( region if region is not None else os.environ.get("AWS_DEFAULT_REGION") ) self.notification = self._setup_notification(notification) + self.kwargs = kwargs # init directed graph dict where values are a set. # we do it like this so that we can use toposort. self.directed_graph = defaultdict(set) @@ -138,6 +139,7 @@ def build_workflow(self): definition=self.chain_of_tasks, role=self.role.role_arn, client=sfn_client, + **self.kwargs, ) def create(self): @@ -151,6 +153,7 @@ def create(self): state_machine_name=self.unique_name, role_arn=self.role.role_arn, definition_string=cfn_template, + **self.kwargs, ) def _setup_notification(