diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py index f40c2c2bc5..7f9b75b367 100644 --- 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 index 6aa02d7983..29a409f529 100644 --- 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