diff options
| author | Taku Kudo <taku@google.com> | 2018-09-01 04:10:01 +0300 |
|---|---|---|
| committer | Taku Kudo <taku@google.com> | 2018-09-01 04:10:01 +0300 |
| commit | f617fcb697eaeaa09b7b4d73559def57f042194e (patch) | |
| tree | d7c28921348ceadc478b0118d966bc32686c0821 /src/char_model_trainer_test.cc | |
| parent | 9f63c57e4b51952b236672a17d6d8c15f25c3d68 (diff) | |
Introduced new Filesystem API.
Diffstat (limited to 'src/char_model_trainer_test.cc')
| -rw-r--r-- | src/char_model_trainer_test.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/char_model_trainer_test.cc b/src/char_model_trainer_test.cc index f628cde..9e7fe27 100644 --- a/src/char_model_trainer_test.cc +++ b/src/char_model_trainer_test.cc @@ -15,6 +15,7 @@ #include "char_model_trainer.h" #include <string> #include <vector> +#include "filesystem.h" #include "sentencepiece_processor.h" #include "testharness.h" #include "util.h" @@ -32,9 +33,9 @@ std::string RunTrainer(const std::vector<std::string> &input, int size) { const std::string input_file = input_scoped_file.filename(); const std::string model_prefix = model_scoped_file.filename(); { - io::OutputBuffer output(input_file); + auto output = filesystem::NewWritableFile(input_file); for (const auto &line : input) { - output.WriteLine(line); + output->WriteLine(line); } } |