From 9df9da34a3280664277f326da573993affbe7be6 Mon Sep 17 00:00:00 2001 From: Ali Zarezade Date: Sat, 3 Feb 2018 17:21:34 +0330 Subject: [PATCH] Fix init_model issue Fixing issue #1928 --- spacy/cli/init_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/cli/init_model.py b/spacy/cli/init_model.py index 6e3369f4d..99a6e87eb 100644 --- a/spacy/cli/init_model.py +++ b/spacy/cli/init_model.py @@ -36,7 +36,7 @@ def init_model(lang, output_dir, freqs_loc, clusters_loc=None, vectors_loc=None, vectors_loc = ensure_path(vectors_loc) probs, oov_prob = read_freqs(freqs_loc) - vectors_data, vector_keys = read_vectors(vectors_loc) if vectors_loc else None, None + vectors_data, vector_keys = read_vectors(vectors_loc) if vectors_loc else (None, None) clusters = read_clusters(clusters_loc) if clusters_loc else {} nlp = create_model(lang, probs, oov_prob, clusters, vectors_data, vector_keys, prune_vectors) @@ -69,7 +69,7 @@ def create_model(lang, probs, oov_prob, clusters, vectors_data, vector_keys, pru lex_added += 1 nlp.vocab.cfg.update({'oov_prob': oov_prob}) - if vectors_data: + if len(vectors_data): nlp.vocab.vectors = Vectors(data=vectors_data, keys=vector_keys) if prune_vectors >= 1: nlp.vocab.prune_vectors(prune_vectors)