protobuf semantics differs on py3.9

Signed-off-by: JackUrb <[email protected]>
flyteorg · Aug 15, 2024 · c411fc7 · c411fc7
1 parent 19d231e
commit c411fc7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/flytekit/core/type_engine.py b/flytekit/core/type_engine.py
@@ -718,9 +718,12 @@ def to_literal(self, ctx: FlyteContext, python_val: T, python_type: Type[T], exp
         struct = Struct()
         try:
             message_dict = _MessageToDict(cast(Message, python_val))
-            if isinstance(message_dict, list):
-                return self._handle_list_literal(ctx, message_dict)
-            struct.update(message_dict)
+            try:
+                struct.update(message_dict)
+            except Exception:
+                if isinstance(message_dict, list):
+                    return self._handle_list_literal(ctx, message_dict)
+                raise
         except Exception:
             raise TypeTransformerFailedError("Failed to convert to generic protobuf struct")
         return Literal(scalar=Scalar(generic=struct))