diff --git a/torch_geometric/utils/rag/graph_store.py b/torch_geometric/utils/rag/graph_store.py index 7a17cf4ef7a1..4d5fcbc9418c 100644 --- a/torch_geometric/utils/rag/graph_store.py +++ b/torch_geometric/utils/rag/graph_store.py @@ -1,6 +1,7 @@ from typing import Optional, Union from torch import Tensor + from torch_geometric.data import FeatureStore from torch_geometric.distributed import LocalGraphStore from torch_geometric.sampler import (