From adaa44884f76bd4c7c2225e8e309132cb9111906 Mon Sep 17 00:00:00 2001 From: Eduardo Apolinario Date: Fri, 18 Feb 2022 14:49:28 -0800 Subject: [PATCH] wip - Add support for univariate lists Signed-off-by: Eduardo Apolinario --- flytekit/core/type_engine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flytekit/core/type_engine.py b/flytekit/core/type_engine.py index e44f3fa737..3354dad84b 100644 --- a/flytekit/core/type_engine.py +++ b/flytekit/core/type_engine.py @@ -803,7 +803,9 @@ def get_literal_type(self, t: Type[T]) -> Optional[LiteralType]: def to_literal(self, ctx: FlyteContext, python_val: T, python_type: Type[T], expected: LiteralType) -> Literal: t = self.get_sub_type(python_type) lit_list = [TypeEngine.to_literal(ctx, x, t, expected.collection_type) for x in python_val] # type: ignore - return Literal(collection=LiteralCollection(literals=lit_list)) + hash = "|".join([literal.hash for literal in lit_list if literal.hash]) + print(f"hash={hash}") + return Literal(collection=LiteralCollection(literals=lit_list), hash=hash if hash != "|" else None) def to_python_value(self, ctx: FlyteContext, lv: Literal, expected_python_type: Type[T]) -> typing.List[T]: st = self.get_sub_type(expected_python_type)