diff --git a/sdk/python/kfp/dsl/_resource_op.py b/sdk/python/kfp/dsl/_resource_op.py index d7f4083dda9..463d5bbdef2 100644 --- a/sdk/python/kfp/dsl/_resource_op.py +++ b/sdk/python/kfp/dsl/_resource_op.py @@ -32,6 +32,7 @@ class Resource(object): "merge_strategy": "str", "success_condition": "str", "failure_condition": "str", + "set_owner_reference": "bool", "manifest": "str", "flags": "list[str]" } @@ -40,6 +41,7 @@ class Resource(object): "merge_strategy": "str", "success_condition": "str", "failure_condition": "str", + "set_owner_reference": "bool", "manifest": "str", "flags": "list[str]" } @@ -48,6 +50,7 @@ class Resource(object): "merge_strategy": "mergeStrategy", "success_condition": "successCondition", "failure_condition": "failureCondition", + "set_owner_reference": "setOwnerReference", "manifest": "manifest", "flags": "flags" } @@ -57,6 +60,7 @@ def __init__(self, merge_strategy: str = None, success_condition: str = None, failure_condition: str = None, + set_owner_reference: bool = None, manifest: str = None, flags: Optional[List[str]] = None): """Create a new instance of Resource""" @@ -64,6 +68,7 @@ def __init__(self, self.merge_strategy = merge_strategy self.success_condition = success_condition self.failure_condition = failure_condition + self.set_owner_reference = set_owner_reference self.manifest = manifest self.flags = flags @@ -100,6 +105,7 @@ def __init__(self, merge_strategy: str = None, success_condition: str = None, failure_condition: str = None, + set_owner_reference: bool = None, attribute_outputs: Optional[Dict[str, str]] = None, flags: Optional[List[str]] = None, **kwargs): @@ -118,7 +124,7 @@ def __init__(self, if merge_strategy and action != "apply": raise ValueError("You can't set merge_strategy when action != 'apply'") - + # if action is delete, there should not be any outputs, success_condition, and failure_condition if action == "delete" and (success_condition or failure_condition or attribute_outputs): raise ValueError("You can't set success_condition, failure_condition, or attribute_outputs when action == 'delete'") @@ -130,6 +136,7 @@ def __init__(self, "merge_strategy": merge_strategy, "success_condition": success_condition, "failure_condition": failure_condition, + "set_owner_reference": set_owner_reference, "flags": flags } # `resource` prop in `io.argoproj.workflow.v1alpha1.Template`