Fix inheritance chain for load_word2vec_format (return correct clas…

…s in case when you create an child class based on kv) (#1968) * fix the inheritance chain * make same change for Poincare
piskvorky · Mar 9, 2018 · 2ee7a2c · 2ee7a2c
1 parent bfb33ff
commit 2ee7a2c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -1115,7 +1115,7 @@ def load_word2vec_format(cls, fname, fvocab=None, binary=False, encoding='utf8',
         """
         # from gensim.models.word2vec import load_word2vec_format
         return _load_word2vec_format(
-            Word2VecKeyedVectors, fname, fvocab=fvocab, binary=binary, encoding=encoding, unicode_errors=unicode_errors,
+            cls, fname, fvocab=fvocab, binary=binary, encoding=encoding, unicode_errors=unicode_errors,
             limit=limit, datatype=datatype)
 
     def get_keras_embedding(self, train_embeddings=False):

diff --git a/gensim/models/poincare.py b/gensim/models/poincare.py
@@ -869,7 +869,7 @@ def load_word2vec_format(cls, fname, fvocab=None, binary=False, encoding='utf8',
 
         """
         return _load_word2vec_format(
-            PoincareKeyedVectors, fname, fvocab=fvocab, binary=binary, encoding=encoding, unicode_errors=unicode_errors,
+            cls, fname, fvocab=fvocab, binary=binary, encoding=encoding, unicode_errors=unicode_errors,
             limit=limit, datatype=datatype)
 
     @staticmethod