Skip to content

Commit

Permalink
pass kwargs
Browse files Browse the repository at this point in the history
  • Loading branch information
vincentclaes committed Aug 10, 2021
1 parent 57c6b2b commit 73c3d59
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions datajob/stepfunctions/stepfunctions_workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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):
Expand All @@ -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(
Expand Down

0 comments on commit 73c3d59

Please sign in to comment.