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

Flyte

<https://github.com/flyteorg/flytekit/pull/1562|#1562 TypeTransformer for TensorFlow model>
Pull request opened by <https://github.com/samhita-alla|samhita-alla>
*TL;DR*

This PR enables `tf.keras.Model` to be utilized as a Flyte type. The code's adapted from <https://github.com/flyteorg/flytekit/pull/1241|#1241>.

*Type*

☐ Bug Fix
☐ Feature
☑︎ Plugin

*Are all requirements met?*

☑︎ Code completed
☐ Smoke tested
☑︎ Unit tests added
☐ Code documentation added
☐ Any pending items have an associated Issue

*Complete description*

_How did you fix the bug, make the feature etc. Link to any design docs etc_

*Tracking Issue*

Fixes <https://github.com/flyteorg/flyte/issues/2570|flyteorg/flyte#2570>

*Follow-up issue*

_NA_  
OR  
_<https://github.com/flyteorg/flyte/issues/|https://github.com/flyteorg/flyte/issues/>_
<https://github.com/flyteorg/flytekit|flyteorg/flytekit>
:white_check_mark: All checks have passed
30/30 successful checks

<https://github.com/flyteorg/flytekit/pull/1562|#1562 TypeTransformer for TensorFlow model>
Pull request merged by <https://github.com/samhita-alla|samhita-alla>
<https://github.com/flyteorg/flytekit|flyteorg/flytekit>