Hi I have a question regarding flyte support for tensorflow

Question

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

Yue Shang · Answer

the tf data Dataset look sth like this

Yue Shang · Answer

``` lt PrefetchDataset element spec= feature1 TensorSpec shape= 32 dtype=tf float32 name=None feature2 TensorSpec shape= 32 dtype=tf float32 name=None gt ```

Kevin Su · Answer

could you share code snippet that will be helpful