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/unigram_model_trainer_test.cc | |
parent | 31153b117294830ab41ff3e9ee4f0a7323f16d8d (diff) |
CHECK to Status migration for Trainer.
Diffstat (limited to 'src/unigram_model_trainer_test.cc')
-rw-r--r-- | src/unigram_model_trainer_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/unigram_model_trainer_test.cc b/src/unigram_model_trainer_test.cc index 096f70e..a88dc71 100644 --- a/src/unigram_model_trainer_test.cc +++ b/src/unigram_model_trainer_test.cc @@ -49,7 +49,7 @@ TEST(UnigramTrainerTest, EndToEndTest) { test::ScopedTempFile sf("tmp_model"); trainer_spec.set_model_prefix(sf.filename()); unigram::Trainer trainer(trainer_spec, normalizer_spec); - trainer.Train(); + EXPECT_OK(trainer.Train()); SentencePieceProcessor sp; EXPECT_OK(sp.Load(std::string(sf.filename()) + ".model")); |