keeping MRs separate.

Signed-off-by: Oswaldo Gomez <[email protected]>
kubeflow · Aug 2, 2024 · 1df116e · 1df116e
1 parent 429d0f3
commit 1df116e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/samples/core/loop_parallelism/loop_parallelism.py b/samples/core/loop_parallelism/loop_parallelism.py
@@ -15,7 +15,7 @@
 from kfp import compiler, dsl
 
 @dsl.component()
-def print_op(s: int):
+def print_op(s: str):
     print(s)
 
 @dsl.pipeline(name='my-pipeline')

diff --git a/samples/core/loop_parameter/loop_parameter.py b/samples/core/loop_parameter/loop_parameter.py
@@ -14,8 +14,9 @@ def concat_op(a: str, b: str) -> str:
 
 
 @dsl.component
-def generate_op() -> list:
-    return [{'a': i, 'b': i * 10} for i in range(1, 5)]
+def generate_op() -> str:
+    import json
+    return json.dumps([{'a': i, 'b': i * 10} for i in range(1, 5)])
 
 @dsl.pipeline(name='pipeline-with-loop-parameter')
 def my_pipeline(