Update backend_utils.py

puririshi98 · web-flow · commit fdde3b7345e2 · 2025-04-15T07:35:38.000-07:00
diff --git a/torch_geometric/utils/rag/backend_utils.py b/torch_geometric/utils/rag/backend_utils.py
@@ -33,7 +33,7 @@
     Partitioner,
 )
 from torch_geometric.nn.nlp import SentenceTransformer
-from torch_geometric.typing import EdgeType, NodeType
+from torch_geometric.typing import EdgeType, NodeType, InputNodes
 
 try:
     from pandas import DataFrame

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@`
`33`	`33`	`Partitioner,`
`34`	`34`	`)`
`35`	`35`	`from torch_geometric.nn.nlp import SentenceTransformer`
`36`		`-from torch_geometric.typing import EdgeType, NodeType`
	`36`	`+from torch_geometric.typing import EdgeType, NodeType, InputNodes`
`37`	`37`
`38`	`38`	`try:`
`39`	`39`	`from pandas import DataFrame`