fix import clusterdata

pyg-team · Oct 10, 2023 · 99ce405 · 99ce405
1 parent 0b7fd84
commit 99ce405
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/distributed/partition.py b/torch_geometric/distributed/partition.py
@@ -7,7 +7,7 @@
 import torch
 
 from torch_geometric.data import Data, HeteroData
-from torch_geometric.loader import ClusterData
+from torch_geometric.loader.cluster import ClusterData
 from torch_geometric.typing import (
     Dict,
     EdgeType,