diff --git a/flytekit-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/LaunchPlanRegistry.scala b/flytekit-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/LaunchPlanRegistry.scala index df5c3b438..ae9e19aca 100644 --- a/flytekit-examples-scala/src/main/scala/org/flyte/examples/flytekitscala/LaunchPlanRegistry.scala +++ b/flytekit-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 index ef4d61245..d69b5242a 100644 --- 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,