New way for overriding transforms in CIFAR10DataModule?

I tried changing the train and test transforms of the CIFAR10DataModule following the tutorial here. While there is no error as of now, I do see the following warning:

“DataModule property train_transforms was deprecated in v1.5 and will be removed in v1.7.”

So what is the recommended way to override the train / val / test transforms with my custom transforms in the future?