Reuse add_metadata_from_type

Signed-off-by: Alice Purcell <[email protected]>
argoproj-labs · Nov 6, 2024 · fa6a37f · fa6a37f
1 parent 2831dfd
commit fa6a37f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/hera/shared/_type_util.py b/src/hera/shared/_type_util.py
@@ -106,10 +106,8 @@ def construct_io_from_annotation(python_name: str, annotation: Any) -> Union[Par
         io = Parameter()
 
     io.name = io.name or python_name
-    if isinstance(io, Parameter) and not io.enum:
-        type_ = unwrap_annotation(annotation)
-        if get_origin(type_) is Literal:
-            io.enum = list(get_args(type_))
+    if isinstance(io, Parameter):
+        add_metadata_from_type(io, annotation)
 
     return io