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