Skip to content

Commit 5be555e

Browse files
authored
Allow overriding of _transition in user step-decorators (Netflix#2651)
1 parent 96b0783 commit 5be555e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

metaflow/task.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ def _exec_step_function(self, step_function, orig_step_func, input_obj=None):
150150
graph_node = self.flow._graph[orig_step_func.name]
151151
out_funcs = [getattr(self.flow, f) for f in graph_node.out_funcs]
152152
if out_funcs:
153+
self.flow._transition = None
153154
if isinstance(fake_next_call_args, dict) and fake_next_call_args:
154155
# Not an empty dictionary -- we use this as arguments for the next
155156
# call

0 commit comments

Comments
 (0)