textaugment | TextAugment : Text Augmentation Library | Natural Language Processing library

I am also trying to do the same. The error occurs because the textaugment function t.random_swap() is supposed to work on Python string objects.

In your code, the function is taking in a Tensor with dtype=string. As of now, tensor objects do not have the same methods as Python strings. Hence, the error code.

Nb. tensorflow_text has some additional APIs to work with such tensors of string types. Albeit, it is limited at the moment to tokenization, checking upper or lower case etc. A long winded workaround is to use the py_function wrapper but this reduces performance. Cheers and hope this helps. I opted not to use textaugment in the end in my use case.

Nbb. tf.strings APIs have a bit more functionalities, such as regex replace etc but it is not complicated enough for your use case of augmentation. Would be helpful to see what others come up with, or if there are future updates to either TF or textaugment.