Fix validation error for retry strategy (#558)

Fix validation error for retry strategy

Signed-off-by: lakhanjindam <[email protected]>

docs/examples/workflows/upstream/retry_script.md

@@ -0,0 +1,55 @@
+# Retry Script
+
+> Note: This example is a replication of an Argo Workflow example in Hera. The upstream example can be [found here](https://github.com/argoproj/argo-workflows/blob/master/examples/retry-script.yaml).
+
+
+
+
+=== "Hera"
+
+    ```python linenums="1"
+    from hera.workflows import Workflow, script, RetryStrategy
+
+
+    @script(image="python:alpine3.6", retry_strategy=RetryStrategy(limit=10), add_cwd_to_sys_path=False)
+    def retry_script():
+        import random
+        import sys
+
+        exit_code = random.choice([0, 1, 1])
+        sys.exit(exit_code)
+
+
+    with Workflow(generate_name="retry-script-", entrypoint="retry-script") as w:
+        retry_script()
+    ```
+
+=== "YAML"
+
+    ```yaml linenums="1"
+    apiVersion: argoproj.io/v1alpha1
+    kind: Workflow
+    metadata:
+      generateName: retry-script-
+    spec:
+      entrypoint: retry-script
+      templates:
+      - name: retry-script
+        retryStrategy:
+          limit: '10'
+        script:
+          command:
+          - python
+          image: python:alpine3.6
+          source: 'import random
+
+            import sys
+
+
+            exit_code = random.choice([0, 1, 1])
+
+            sys.exit(exit_code)
+
+            '
+    ```
+

examples/workflows/upstream/retry-script.upstream.yaml

@@ -0,0 +1,20 @@
+# This example demonstrates the use of retries for a single script.
+apiVersion: argoproj.io/v1alpha1
+kind: Workflow
+metadata:
+  generateName: retry-script-
+spec:
+  entrypoint: retry-script
+  templates:
+  - name: retry-script
+    retryStrategy:
+      limit: "10"
+    script:
+      image: python:alpine3.6
+      command: ["python"]
+      # fail with a 66% probability
+      source: |
+        import random;
+        import sys; 
+        exit_code = random.choice([0, 1, 1]); 
+        sys.exit(exit_code)

examples/workflows/upstream/retry-script.yaml

@@ -0,0 +1,24 @@
+apiVersion: argoproj.io/v1alpha1
+kind: Workflow
+metadata:
+  generateName: retry-script-
+spec:
+  entrypoint: retry-script
+  templates:
+  - name: retry-script
+    retryStrategy:
+      limit: '10'
+    script:
+      command:
+      - python
+      image: python:alpine3.6
+      source: 'import random
+
+        import sys
+
+
+        exit_code = random.choice([0, 1, 1])
+
+        sys.exit(exit_code)
+
+        '

examples/workflows/upstream/retry_script.py

@@ -0,0 +1,14 @@
+from hera.workflows import RetryStrategy, Workflow, script
+
+
+@script(image="python:alpine3.6", retry_strategy=RetryStrategy(limit=10), add_cwd_to_sys_path=False)
+def retry_script():
+    import random
+    import sys
+
+    exit_code = random.choice([0, 1, 1])
+    sys.exit(exit_code)
+
+
+with Workflow(generate_name="retry-script-", entrypoint="retry-script") as w:
+    retry_script()

src/hera/workflows/retry_strategy.py

@@ -36,7 +36,7 @@ class RetryStrategy(_BaseModel):
     affinity: Optional[RetryAffinity] = None
     backoff: Optional[Backoff] = None
     expression: Optional[str] = None
-    limit: Optional[Union[int, str]] = None
+    limit: Optional[Union[str, int, IntOrString]] = None
     retry_policy: Optional[Union[str, RetryPolicy]] = None
 
     @validator("retry_policy", pre=True)
@@ -52,7 +52,7 @@ def _convert_limit(cls, v):
         if v is None or isinstance(v, IntOrString):
             return v
 
-        return IntOrString(__root__=str(v))  # int or str
+        return str(v)  # int or str
 
     def build(self) -> _ModelRetryStrategy:
         return _ModelRetryStrategy(