From 2d856ca2e2f2d8640efcd287b7e8f99855b339f3 Mon Sep 17 00:00:00 2001
From: Tommy Li <Tommy.chaoping.li@ibm.com>
Date: Thu, 28 Mar 2024 11:06:16 -0700
Subject: [PATCH] Update loop_output.py example for the new parallel loop type
 requirement

---
 samples/core/loop_output/loop_output.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/samples/core/loop_output/loop_output.py b/samples/core/loop_output/loop_output.py
index 60148a991c2..a52d2916c50 100644
--- a/samples/core/loop_output/loop_output.py
+++ b/samples/core/loop_output/loop_output.py
@@ -13,11 +13,12 @@
 # limitations under the License.
 
 from kfp import compiler, dsl
+from typing import List
 
 
 @dsl.component
-def args_generator_op() -> str:
-    return '[1.1, 1.2, 1.3]'
+def args_generator_op() -> List[str]:
+    return ['1.1', '1.2', '1.3']
 
 
 # TODO(Bobgy): how can we make this component with type float?