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/bpe_model_trainer_test.cc | |
parent | 9f63c57e4b51952b236672a17d6d8c15f25c3d68 (diff) |
Introduced new Filesystem API.
Diffstat (limited to 'src/bpe_model_trainer_test.cc')
-rw-r--r-- | src/bpe_model_trainer_test.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/bpe_model_trainer_test.cc b/src/bpe_model_trainer_test.cc index 944a963..1f3b08a 100644 --- a/src/bpe_model_trainer_test.cc +++ b/src/bpe_model_trainer_test.cc @@ -16,6 +16,7 @@ #include <string> #include <vector> +#include "filesystem.h" #include "flags.h" #include "sentencepiece_processor.h" #include "sentencepiece_trainer.h" @@ -39,9 +40,9 @@ std::string RunTrainer( 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); } } |