diff --git a/torch_geometric/transforms/__init__.py b/torch_geometric/transforms/__init__.py index fadf348f2423..b9e2c257f930 100644 --- a/torch_geometric/transforms/__init__.py +++ b/torch_geometric/transforms/__init__.py @@ -112,5 +112,5 @@ from torch_geometric.deprecation import deprecated # noqa -NeighborTranslate = deprecated("use 'transforms.RandomJitter' instead", - 'transforms.RandomTranslate')(RandomJitter) +RandomTranslate = deprecated("use 'transforms.RandomJitter' instead", + 'transforms.RandomTranslate')(RandomJitter)