diff --git a/torch_geometric/utils/rag/backend_utils.py b/torch_geometric/utils/rag/backend_utils.py index e8e3b941a423..433445161d2f 100644 --- a/torch_geometric/utils/rag/backend_utils.py +++ b/torch_geometric/utils/rag/backend_utils.py @@ -129,7 +129,6 @@ def load(self, pid: Optional[int] = None) -> RemoteGraphBackend: graph_store = self.graph_store_type.from_data( edge_id=data_obj['edge_id'], edge_index=data_obj.edge_index, num_nodes=data_obj.num_nodes, is_sorted=True) - graph_store.edge_index = graph_store.edge_index.contiguous() feature_store = self.feature_store_type.from_data( node_id=data_obj['node_id'], x=data_obj.x, edge_id=data_obj['edge_id'], edge_attr=data_obj.edge_attr)