From 502659aeaa7dcf1e5dadb7df7d813d346e663878 Mon Sep 17 00:00:00 2001 From: xnuohz Date: Tue, 7 Jan 2025 23:58:03 +0800 Subject: [PATCH] update --- torch_geometric/datasets/tag_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch_geometric/datasets/tag_dataset.py b/torch_geometric/datasets/tag_dataset.py index 557eaeeffb6b..0cfe912a1b5f 100644 --- 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