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

Flyte

Hi! I have a question regarding flyte support for tensorflow data. I have generated a `tf.data.Dataset` in one of the tasks and return it as input for next task, but got data type error (when pickle serialize this data)`tensorflow.python.framework.errors_impl.InvalidArgumentError'&gt;: Cannot convert a Tensor of dtype variant to a NumPy array` , I'm kind of new to both tensorflow and flyte, so wondering is there any guidance on how to resolve this issue?

```&lt;PrefetchDataset element_spec=({'feature1': TensorSpec(shape=(32,), dtype=tf.float32, name=None), 'feature2': TensorSpec(shape=(32,), dtype=tf.float32, name=None)})&gt;```

could you share code snippet. that will be helpful