Merge pull request #28 from JuliaText/de/keepwords

make sure keep_words is a set
JuliaText · Nov 22, 2019 · 65390f1 · 65390f1 · oxinabox · Nov 22, 2019
2 parents 6232141 + e2dad12
commit 65390f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Embeddings"
 uuid = "c5bfea45-b7f1-5224-a596-15500f5db411"
 authors = ["Lyndon White"]
-version = "0.4.0"
+version = "0.4.1"
 
 [deps]
 AutoHashEquals = "15f4f7f2-30c1-5605-9d31-71845cf9641f"

diff --git a/src/Embeddings.jl b/src/Embeddings.jl
@@ -96,7 +96,7 @@ function load_embeddings(::Type{T},
         max_vocab_size=typemax(Int),
         keep_words=Set()) where T<:EmbeddingSystem
 
-    EmbeddingTable(_load_embeddings(T, embedding_file, max_vocab_size,  keep_words)...)
+    EmbeddingTable(_load_embeddings(T, embedding_file, max_vocab_size, Set(keep_words))...)
 end
 
 end