Can not import "AddTrainValTestMask" class from "torch_geometric.transforms"

[zhihuat]

Hi! Thanks for the fantastic tool first.
I met a problem when importing "AddTrainValTestMask" from "torch_geometric.transforms". The source code shows that it has been deleted in the latest version. Is there has any alternative Class to use?
Thank you in advance!

[rusty1s]

It got renamed to RandomNodeSplit :)

[zhihuat]

Thank you very much! I replaced the "AddTrainValTestMask " with "RandomNodeSplit" It works great!