Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Flyte

<https://github.com/flyteorg/flyte-sdk/pull/170|#170 Support Literal type (as enums) and Enums in dataclasses and pydantic models>
Pull request opened by <https://github.com/kumare3|kumare3>
Intensity = Literal["low", "medium", "high"]

@env.task
async def literal_echo(i: Intensity) -&gt; Intensity:
    return i


@env.task
async def literal_task(i: Intensity) -&gt; str:
    return f"Intensity is {i}"

Works correctly
<https://github.com/flyteorg/flyte-sdk|flyteorg/flyte-sdk>

<https://github.com/flyteorg/flyte-sdk/pull/170|#170 Support Literal type (as enums) and Enums in dataclasses and pydantic models>
Pull request merged by <https://github.com/kumare3|kumare3>
<https://github.com/flyteorg/flyte-sdk|flyteorg/flyte-sdk>