update

pyg-team · Jan 7, 2025 · 502659a · 502659a
1 parent 74f2bb9
commit 502659a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_geometric/datasets/tag_dataset.py b/torch_geometric/datasets/tag_dataset.py
@@ -190,7 +190,8 @@ def token(self) -> Dict[str, Tensor]:
     @property
     def llm_explanation_token(self) -> Dict[str, Tensor]:
         if self._llm_explanation_token is None:  # lazy load
-            self._llm_explanation_token = self.tokenize_graph()
+            self._llm_explanation_token = self.tokenize_graph(
+                text_type='llm_explanation')
         return self._llm_explanation_token
 
     # load is_gold after init