IT

Signed-off-by: Hongxin Liang <honnix@users.noreply.github.com>
flyteorg · Nov 22, 2023 · 57a581c · 57a581c
1 parent e53f361
commit 57a581c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/...t-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/LaunchPlanRegistry.scala b/...t-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/LaunchPlanRegistry.scala
@@ -73,14 +73,19 @@ class LaunchPlanRegistry extends SimpleSdkLaunchPlanRegistry {
             6.toDouble,
             "hello",
             List("1", "2"),
-            List(NestedNested(7.toDouble, NestedNestedNested("world"))),
+            List(NestedNested(7.toDouble, Some(NestedNestedNested("world")))),
             Map("1" -> "1", "2" -> "2"),
-            Map("foo" -> NestedNested(7.toDouble, NestedNestedNested("world"))),
+            Map(
+              "foo" -> NestedNested(
+                7.toDouble,
+                Some(NestedNestedNested("world"))
+              )
+            ),
             Some(false),
             None,
             Some(List("3", "4")),
             Some(Map("3" -> "3", "4" -> "4")),
-            NestedNested(7.toDouble, NestedNestedNested("world"))
+            NestedNested(7.toDouble, Some(NestedNestedNested("world")))
           )
         )
       )

diff --git a/flytekit-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/NestedIOTask.scala b/flytekit-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/NestedIOTask.scala
@@ -24,7 +24,7 @@ import org.flyte.flytekitscala.{
 }
 
 case class NestedNestedNested(string: String)
-case class NestedNested(double: Double, nested: NestedNestedNested)
+case class NestedNested(double: Double, nested: Option[NestedNestedNested])
 case class Nested(
     boolean: Boolean,
     byte: Byte,