Hey @broad-monitor-993! Regarding https://github.com/flyteorg/flyte/issues/2759,

tf.Module

is the parent class for

tf.keras.layers.Layer

and

tf.keras.Model

(see the note on https://www.tensorflow.org/guide/intro_to_modules page). So we need a type transformer for

tf.Module

. But we have two PRs — https://github.com/flyteorg/flytekit/pull/1242 to fix https://github.com/flyteorg/flyte/issues/2759 and https://github.com/flyteorg/flytekit/pull/1241 to fix https://github.com/flyteorg/flyte/issues/2570 that have the same functionality — a type transformer subclassing

tf.keras.Model

. We’ll need to tell one of them to modify the code. So which PR shall we retain?