update example

kubeflow · Mar 27, 2024 · c387009 · c387009
1 parent 94b6b7e
commit c387009
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/samples/core/loop_parameter/loop_parameter.py b/samples/core/loop_parameter/loop_parameter.py
@@ -1,4 +1,5 @@
 from kfp import compiler, dsl
+from typing import List
 
 
 @dsl.component
@@ -14,7 +15,7 @@ def concat_op(a: str, b: str) -> str:
 
 
 @dsl.component
-def generate_op() -> str:
+def generate_op() -> List[str]:
     import json
     return json.dumps([{'a': i, 'b': i * 10} for i in range(1, 5)])