Problem with import of class

1

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!

2

I met the same problem, have you fixed it?

3

Update torch geometric in your environment

4

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