fix pyright

mosaicml · Oct 27, 2023 · 5e688e7 · 5e688e7
1 parent e994fed
commit 5e688e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/tokenizers/tiktoken.py b/llmfoundry/tokenizers/tiktoken.py
@@ -63,7 +63,7 @@ def pickle_Encoding(enc: tiktoken.Encoding):
                                       mergeable_ranks=enc._mergeable_ranks,
                                       special_tokens=enc._special_tokens), ())
 
-        copyreg.pickle(tiktoken.core.Encoding, pickle_Encoding)
+        copyreg.pickle(tiktoken.Encoding, pickle_Encoding)
 
         if model_name is not None and encoding_name is not None:
             raise ValueError(