Export alignments from python

browsermt · Feb 15, 2022 · 21f4c50 · 21f4c50
1 parent 46a160b
commit 21f4c50
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bindings/python/bergamot.cpp b/bindings/python/bergamot.cpp
@@ -172,6 +172,10 @@ PYBIND11_MODULE(_bergamot, m) {
   py::bind_vector<std::vector<std::string>>(m, "VectorString");
   py::bind_vector<std::vector<Response>>(m, "VectorResponse");
 
+  py::bind_vector<std::vector<float>>(m, "VectorFloat");
+  py::bind_vector<Alignment>(m, "Alignment");
+  py::bind_vector<Alignments>(m, "Alignments");
+
   py::enum_<ConcatStrategy>(m, "ConcatStrategy")
       .value("FAITHFUL", ConcatStrategy::FAITHFUL)
       .value("SPACE", ConcatStrategy::SPACE)