Add getters for options and subword encoder

OpenNMT · Oct 26, 2020 · 46868ea · 46868ea
1 parent 5e108a9
commit 46868ea
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/include/onmt/Tokenizer.h b/include/onmt/Tokenizer.h
@@ -98,7 +98,12 @@ namespace onmt
 
     void set_subword_encoder(const std::shared_ptr<const SubwordEncoder>& subword_encoder);
 
-    const Options& options() const
+    const std::shared_ptr<const SubwordEncoder>& get_subword_encoder() const
+    {
+      return _subword_encoder;
+    }
+
+    const Options& get_options() const
     {
       return _options;
     }