Problem with import of class

I would like to import a class from torch_geometric, but I keep receiving the error. I use the following line of code: from torch_geometric.transforms import AddTrainValTestMask
The error is: ImportError: cannot import name 'AddTrainValTestMask' from 'torch_geometric.transforms'

Documentation you may find here: torch_geometric.transforms — pytorch_geometric 1.6.3 documentation

Thank you!

I met the same problem, have you fixed it?

Update torch geometric in your environment

It seems this class has been removed from version 2.
Is there any class that has the same function?