fix(tokenize): fix wrong lifetime of build_from_trie

ModelTC · Nov 6, 2023 · a1bb23c · a1bb23c
1 parent 14c7761
commit a1bb23c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/tokenize.rs b/src/utils/tokenize.rs
@@ -120,10 +120,10 @@ impl<'s, TransTable: TransitionTable, TokenIDType: Clone + Default + PartialEq>
 pub mod trie {
     use crate::{GeneralSAM, TransitionTable, Trie, TrieNodeAlike, TrieNodeID, TrieState};
 
-    impl<'s, TransTable: TransitionTable> super::GreedyTokenizer<'s, TransTable, TrieNodeID> {
+    impl<'s, 't, TransTable: TransitionTable> super::GreedyTokenizer<'s, TransTable, TrieNodeID> {
         pub fn build_from_trie<TT: TransitionTable<KeyType = TransTable::KeyType>>(
             sam: &'s GeneralSAM<TransTable>,
-            trie_state: TrieState<'s, TT>,
+            trie_state: TrieState<'t, TT>,
         ) -> Self {
             Self::build(sam, trie_state, |tn| tn.node_id)
         }