diff options
Diffstat (limited to 'python/src/sentencepiece/__init__.py')
-rw-r--r-- | python/src/sentencepiece/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/python/src/sentencepiece/__init__.py b/python/src/sentencepiece/__init__.py index 001ffc7..566f810 100644 --- a/python/src/sentencepiece/__init__.py +++ b/python/src/sentencepiece/__init__.py @@ -370,6 +370,9 @@ class SentencePieceProcessor(object): # Register SentencePieceProcessor in _sentencepiece: _sentencepiece.SentencePieceProcessor_swigregister(SentencePieceProcessor) + +def SetRandomGeneratorSeed(seed): + return _sentencepiece.SetRandomGeneratorSeed(seed) class SentencePieceTrainer(object): thisown = property(lambda x: x.this.own(), lambda x, v: x.this.own(v), doc="The membership flag") @@ -516,6 +519,7 @@ for m in [ _add_snake_case(SentencePieceProcessor) _add_snake_case(SentencePieceTrainer) +set_random_generator_seed = SetRandomGeneratorSeed |