Add Embeddings::write for saving embeddings

Author: danieldk

Cargo.toml

@@ -14,5 +14,5 @@ features = ["extension-module"]
 
 [dependencies]
 failure = "0.1"
-rust2vec = "0.5"
+rust2vec = "0.5.1"
 toml = "0.4"

src/lib.rs

@@ -2,7 +2,7 @@
 
 use std::cell::RefCell;
 use std::fs::File;
-use std::io::BufReader;
+use std::io::{BufReader, BufWriter};
 use std::rc::Rc;
 
 use failure::Error;
@@ -237,6 +237,24 @@ impl PyEmbeddings {
 
         Ok(r)
     }
+
+    /// Write the embeddings to a finalfusion file.
+    fn write(&self, filename: &str) -> PyResult<()> {
+        let f = File::create(filename)?;
+        let mut writer = BufWriter::new(f);
+
+        let embeddings = self.embeddings.borrow();
+
+        use EmbeddingsWrap::*;
+        match &*embeddings {
+            View(e) => e
+                .write_embeddings(&mut writer)
+                .map_err(|err| exceptions::IOError::py_err(err.to_string())),
+            NonView(e) => e
+                .write_embeddings(&mut writer)
+                .map_err(|err| exceptions::IOError::py_err(err.to_string())),
+        }
+    }
 }
 
 #[pyproto]