Skip to content

Commit 494b464

Browse files
Copilotxenova
andcommitted
Fix exports to match Python API: remove duplicates, add base classes, use correct short names
Co-authored-by: xenova <[email protected]>
1 parent cb66878 commit 494b464

File tree

5 files changed

+28
-29
lines changed

5 files changed

+28
-29
lines changed

src/decoders.ts

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
1-
export { default as BPEDecoder } from "./core/decoder/BPEDecoder";
1+
export { default as Decoder } from "./core/Decoder";
2+
export { default as ByteLevel } from "./core/decoder/ByteLevelDecoder";
3+
export { default as Replace } from "./core/decoder/ReplaceDecoder";
4+
export { default as WordPiece } from "./core/decoder/WordPieceDecoder";
25
export { default as ByteFallback } from "./core/decoder/ByteFallback";
3-
export { default as ByteLevelDecoder } from "./core/decoder/ByteLevelDecoder";
4-
export { default as CTCDecoder } from "./core/decoder/CTCDecoder";
5-
export { default as DecoderSequence } from "./core/decoder/DecoderSequence";
6-
export { default as FuseDecoder } from "./core/decoder/FuseDecoder";
7-
export { default as MetaspaceDecoder } from "./core/decoder/MetaspaceDecoder";
8-
export { default as ReplaceDecoder } from "./core/decoder/ReplaceDecoder";
9-
export { default as StripDecoder } from "./core/decoder/StripDecoder";
10-
export { default as WordPieceDecoder } from "./core/decoder/WordPieceDecoder";
6+
export { default as Fuse } from "./core/decoder/FuseDecoder";
7+
export { default as Strip } from "./core/decoder/StripDecoder";
8+
export { default as Metaspace } from "./core/decoder/MetaspaceDecoder";
9+
export { default as BPEDecoder } from "./core/decoder/BPEDecoder";
10+
export { default as CTC } from "./core/decoder/CTCDecoder";
11+
export { default as Sequence } from "./core/decoder/DecoderSequence";

src/models.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1+
export { default as Model } from "./core/TokenizerModel";
12
export { default as BPE } from "./core/tokenizerModelImplementations/BPE";
2-
export { default as LegacyTokenizerModel } from "./core/tokenizerModelImplementations/LegacyTokenizerModel";
33
export { default as Unigram } from "./core/tokenizerModelImplementations/Unigram";
44
export { default as WordPiece } from "./core/tokenizerModelImplementations/WordPieceTokenizer";
5-
export { default as WordPieceTokenizer } from "./core/tokenizerModelImplementations/WordPieceTokenizer";

src/normalizers.ts

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
1+
export { default as Normalizer } from "./core/Normalizer";
12
export { default as BertNormalizer } from "./core/normalizer/BertNormalizer";
2-
export { default as Lowercase } from "./core/normalizer/Lowercase";
3-
export { default as NFC } from "./core/normalizer/NFC";
43
export { default as NFD } from "./core/normalizer/NFD";
5-
export { default as NFKC } from "./core/normalizer/NFKC";
64
export { default as NFKD } from "./core/normalizer/NFKD";
7-
export { default as NormalizerSequence } from "./core/normalizer/NormalizerSequence";
8-
export { default as Precompiled } from "./core/normalizer/Precompiled";
5+
export { default as NFC } from "./core/normalizer/NFC";
6+
export { default as NFKC } from "./core/normalizer/NFKC";
7+
export { default as Sequence } from "./core/normalizer/NormalizerSequence";
8+
export { default as Lowercase } from "./core/normalizer/Lowercase";
99
export { default as Prepend } from "./core/normalizer/Prepend";
10-
export { default as Replace } from "./core/normalizer/Replace";
10+
export { default as Strip } from "./core/normalizer/StripNormalizer";
1111
export { default as StripAccents } from "./core/normalizer/StripAccents";
12-
export { default as StripNormalizer } from "./core/normalizer/StripNormalizer";
13-
export { default as UnicodeNormalizer } from "./core/normalizer/UnicodeNormalizer";
12+
export { default as Precompiled } from "./core/normalizer/Precompiled";
13+
export { default as Replace } from "./core/normalizer/Replace";

src/post-processors.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
1+
export { default as PostProcessor } from "./core/PostProcessor";
12
export { default as BertProcessing } from "./core/postProcessor/BertProcessing";
2-
export { default as ByteLevelPostProcessor } from "./core/postProcessor/ByteLevelPostProcessor";
3-
export { default as PostProcessorSequence } from "./core/postProcessor/PostProcessorSequence";
3+
export { default as ByteLevel } from "./core/postProcessor/ByteLevelPostProcessor";
44
export { default as RobertaProcessing } from "./core/postProcessor/RobertaProcessing";
5+
export { default as Sequence } from "./core/postProcessor/PostProcessorSequence";
56
export { default as TemplateProcessing } from "./core/postProcessor/TemplateProcessing";

src/pre-tokenizers.ts

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,10 @@
1+
export { default as PreTokenizer } from "./core/PreTokenizer";
12
export { default as BertPreTokenizer } from "./core/preTokenizer/BertPreTokenizer";
2-
export { default as ByteLevelPreTokenizer } from "./core/preTokenizer/ByteLevelPreTokenizer";
3-
export { default as DigitsPreTokenizer } from "./core/preTokenizer/DigitsPreTokenizer";
3+
export { default as ByteLevel } from "./core/preTokenizer/ByteLevelPreTokenizer";
4+
export { default as Digits } from "./core/preTokenizer/DigitsPreTokenizer";
45
export { default as Metaspace } from "./core/preTokenizer/MetaspacePreTokenizer";
5-
export { default as MetaspacePreTokenizer } from "./core/preTokenizer/MetaspacePreTokenizer";
6-
export { default as PreTokenizerSequence } from "./core/preTokenizer/PreTokenizerSequence";
7-
export { default as PunctuationPreTokenizer } from "./core/preTokenizer/PunctuationPreTokenizer";
8-
export { default as ReplacePreTokenizer } from "./core/preTokenizer/ReplacePreTokenizer";
9-
export { default as SplitPreTokenizer } from "./core/preTokenizer/SplitPreTokenizer";
6+
export { default as Punctuation } from "./core/preTokenizer/PunctuationPreTokenizer";
7+
export { default as Sequence } from "./core/preTokenizer/PreTokenizerSequence";
8+
export { default as Split } from "./core/preTokenizer/SplitPreTokenizer";
109
export { default as Whitespace } from "./core/preTokenizer/WhitespacePreTokenizer";
11-
export { default as WhitespacePreTokenizer } from "./core/preTokenizer/WhitespacePreTokenizer";
1210
export { default as WhitespaceSplit } from "./core/preTokenizer/WhitespaceSplit";

0 commit comments

Comments
 (0)