fix enum type assertion with python versions less than 3.12 (#2873)

Signed-off-by: Daniel Sola <daniel.sola@union.ai>

flytekit/core/type_engine.py

@@ -937,7 +937,7 @@ def assert_type(self, t: Type[enum.Enum], v: T):
             return
 
         val = v.value if isinstance(v, enum.Enum) else v
-        if val not in t:
+        if val not in [t_item.value for t_item in t]:
             raise TypeTransformerFailedError(f"Value {v} is not in Enum {t}")
 
 

tests/flytekit/unit/core/test_enum_type.py

@@ -0,0 +1,22 @@
+from enum import Enum
+
+from flytekit import task, workflow
+
+
+def test_dynamic_local():
+    class Color(Enum):
+        RED = 'red'
+        GREEN = 'green'
+        BLUE = 'blue'
+
+    @task
+    def my_task(c: Color) -> Color:
+        print(c)
+        return c
+
+    @workflow
+    def wf(c: Color) -> Color:
+        return my_task(c=c)
+
+    res = wf(c=Color.RED)
+    assert res == Color.RED