Feature Request: Add Grain support for keras.layers.TextVectorization

[alenowak]

Keras version: 3.13.2

Currently, keras.layers.TextVectorization.adapt does not support Grain datasets.

ValueError: Attempt to convert a value (<grain._src.python.dataset.transformations.map.MapIterDataset object at 0x11d4cb260>) with an unsupported type (<class 'grain._src.python.dataset.transformations.map.MapIterDataset'>) to a Tensor.

I propose adding Grain support for TextVectorization layer. This would remove another TensorFlow dependency as well.

[dataCenter430]

Hi, @dhantule
I'm going to contribute to Keras actively, since I noticed it's good deep learning framework.
may I pick this up, if needed?
Please assign it to me.
Best regards

[divyashreepathihalli]

Hi @dataCenter430 - I am assigning this to you!

[dataCenter430]

Hi, @divyashreepathihalli thanks for your assignment. 👍
I'm on it and will open a PR soon.
Best regards