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 6
Thank you!