Skip to content

Instantly share code, notes, and snippets.

@EngHabu
Created June 15, 2020 18:57
Show Gist options
  • Save EngHabu/96f06235ea3a5b64ca89fb4814b91d34 to your computer and use it in GitHub Desktop.
Save EngHabu/96f06235ea3a5b64ca89fb4814b91d34 to your computer and use it in GitHub Desktop.
Example for how to override node timeout in FlyteKit
def with_timeout(node, time_out):
"""
:param SdkNode node: The node to override timeout for.
:param datetime.timedelta time_out: New Timeout for the node.
"""
node.metadata._timeout = time_out
return node
@workflow_class
class MyWorkflow(object):
n = with_timeout(my_launch_plan(input1="hello"), datetime.timedelta(days=3))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment