From d48a5563aa4a3481258d003a30610624bc05edd8 Mon Sep 17 00:00:00 2001
From: Colin <>
Date: Mon, 18 Aug 2025 13:03:32 +0800
Subject: [PATCH] Fix get_nodetree.

---
 wit/meaning/meaning_dataset.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/wit/meaning/meaning_dataset.py b/wit/meaning/meaning_dataset.py
index cc3e53c..7d7cfda 100644
--- a/wit/meaning/meaning_dataset.py
+++ b/wit/meaning/meaning_dataset.py
@@ -246,7 +246,7 @@ class MeaningMap:
         start = self.ms_start[meaning]
         seq = self.ms_data[start : start + self.ms_len[meaning]]
 
-        get_tree_node(self, meaning, seq, self.vocab_size, root, seqlist, 0)
+        get_tree_node(self, meaning, seq, self.normal_vocab, root, seqlist, 0)
         root.seq_node = seqlist
         return root
 
@@ -545,6 +545,7 @@ if __name__ == "__main__":
         min_subitem=2,
         max_subitem=6,
         vocab_size=32,
+        reserve_vocab=1,
         size=1024,
         stride=2,
         with_tree=False,