diff options
| author | Taku Kudo <taku@google.com> | 2018-05-06 18:51:20 +0300 |
|---|---|---|
| committer | Taku Kudo <taku@google.com> | 2018-05-06 18:51:20 +0300 |
| commit | cf0eb82d65850172a4661e47668938ab82cb2c76 (patch) | |
| tree | fa6414db5714c6fd15f725d5a9e28cbcb14e20c2 /src/bpe_model_trainer_test.cc | |
| parent | 31153b117294830ab41ff3e9ee4f0a7323f16d8d (diff) | |
CHECK to Status migration for Trainer.
Diffstat (limited to 'src/bpe_model_trainer_test.cc')
| -rw-r--r-- | src/bpe_model_trainer_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/bpe_model_trainer_test.cc b/src/bpe_model_trainer_test.cc index 2336dd5..0610ae5 100644 --- a/src/bpe_model_trainer_test.cc +++ b/src/bpe_model_trainer_test.cc @@ -51,7 +51,7 @@ std::string RunTrainer(const std::vector<std::string> &input, int size) { trainer.Train(); SentencePieceProcessor processor; - processor.Load(model_prefix + ".model"); + EXPECT_OK(processor.Load(model_prefix + ".model")); const auto &model = processor.model_proto(); std::vector<std::string> pieces; |