diff --git a/sdk/python/kfp/dsl/_container_op.py b/sdk/python/kfp/dsl/_container_op.py
index acb070111713..f48c13c28f45 100644
--- a/sdk/python/kfp/dsl/_container_op.py
+++ b/sdk/python/kfp/dsl/_container_op.py
@@ -40,6 +40,7 @@
     'Always',
     'OnError',
     'OnFailure',
+    'OnTransientError',
 )
 
 # Shorthand for PipelineContainerSpec