From fb959cb293ce6d0bedd554fb589c29ae0652f531 Mon Sep 17 00:00:00 2001 From: Rishi Puri Date: Tue, 4 Feb 2025 21:03:08 -0800 Subject: [PATCH] fix --- torch_geometric/loader/rag_loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/torch_geometric/loader/rag_loader.py b/torch_geometric/loader/rag_loader.py index 3e00247384e9..3fe9e4b8b6d8 100644 --- a/torch_geometric/loader/rag_loader.py +++ b/torch_geometric/loader/rag_loader.py @@ -96,10 +96,11 @@ def __init__(self, data: Tuple[RAGFeatureStore, RAGGraphStore], (Default: :obj:`2`). """ fstore, gstore = data - assert len(raw_docs) == len( - embedded_docs), "Need raw and embedded docs to match" self.raw_docs = raw_docs self.k_for_docs = k_for_docs + if self.raw_docs: + assert len(raw_docs) == len( + embedded_docs), "Need raw and embedded docs to match" self.embedded_docs = embedded_docs self.feature_store = fstore self.graph_store = gstore