Merge pull request #317 from uclmr/sr_branch

fix to instantiate memory map embeddings with the right format string
uclnlp · Oct 25, 2017 · 6e95339 · 6e95339
2 parents e2a5d31 + bf16476
commit 6e95339
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jack/io/embeddings/memory_map.py b/jack/io/embeddings/memory_map.py
@@ -26,7 +26,7 @@ def load_memory_map_dir(directory: str) -> Embeddings:
     shape = tuple(meta['shape'])
     vocab = meta['vocab']
     mem_map = np.memmap(mem_map_file, dtype='float32', mode='r+', shape=shape)
-    result = Embeddings(vocab, mem_map, filename=directory, emb_format="mem_map_dir")
+    result = Embeddings(vocab, mem_map, filename=directory, emb_format="memory_map_dir")
     return result