fixed saving bug #91

docs/conf.py

@@ -29,7 +29,7 @@
 author = 'Dimo Angelov'
 
 # The full version, including alpha/beta/rc tags
-release = '1.0.18'
+release = '1.0.19'
 
 
 # -- General configuration ---------------------------------------------------

setup.py

@@ -6,7 +6,7 @@
 setuptools.setup(
     name="top2vec",
     packages=["top2vec"],
-    version="1.0.18",
+    version="1.0.19",
     author="Dimo Angelov",
     author_email="dimo.angelov@gmail.com",
     description="Top2Vec learns jointly embedded topic, document and word vectors.",

top2vec/Top2Vec.py

@@ -402,7 +402,7 @@ def save(self, file):
             self.document_index = None
 
         # serialize word index so that it can be saved
-        if self.word_indexed:
+        if self.words_indexed:
             temp = tempfile.NamedTemporaryFile(mode='w+b')
             self.word_index.save_index(temp.name)
             self.serialized_word_index = temp.read()

top2vec/__init__.py

@@ -1,3 +1,3 @@
 from top2vec.Top2Vec import Top2Vec
 
-__version__ = '1.0.18'
+__version__ = '1.0.19'

top2vec/tests/test_top2vec.py

@@ -2,6 +2,7 @@
 from top2vec import Top2Vec
 from sklearn.datasets import fetch_20newsgroups
 import numpy as np
+import tempfile
 
 # get 20 newsgroups data
 newsgroups_train = fetch_20newsgroups(subset='all', remove=('headers', 'footers', 'quotes'))
@@ -457,3 +458,11 @@ def test_similar_words_index(top2vec_model):
 
     # check that words are returned in decreasing order
     assert all(word_scores[i] >= word_scores[i + 1] for i in range(len(word_scores) - 1))
+
+
+@pytest.mark.parametrize('top2vec_model', models)
+def test_similar_words_index(top2vec_model):
+    temp = tempfile.NamedTemporaryFile(mode='w+b')
+    top2vec_model.save(temp.name)
+    Top2Vec.load(temp.name)
+    temp.close()