Throw exception if load() is called on instance rather than the class

gensim/models/doc2vec.py

@@ -55,6 +55,8 @@
 from six.moves import xrange, zip
 from six import string_types, integer_types, itervalues
 
+from types import MethodType
+
 logger = logging.getLogger(__name__)
 
 try:
@@ -608,6 +610,10 @@ def __init__(self, documents=None, size=300, alpha=0.025, window=8, min_count=5,
         self.dbow_words = dbow_words
         self.dm_concat = dm_concat
         self.dm_tag_count = dm_tag_count
+
+        self.load = methodize(load, self)
+        self.load_word2vec_format = methodize(load, self)
+
         if self.dm and self.dm_concat:
             self.layer1_size = (self.dm_tag_count + (2 * self.window)) * self.vector_size
         else:

gensim/models/word2vec.py

@@ -94,6 +94,7 @@
 from six import iteritems, itervalues, string_types
 from six.moves import xrange
 from types import GeneratorType
+from types import MethodType
 
 logger = logging.getLogger(__name__)
 
@@ -452,6 +453,9 @@ def __init__(
         self.total_train_time = 0
         self.sorted_vocab = sorted_vocab
         self.batch_words = batch_words
+
+        self.load = methodize(load, self)
+        self.load_word2vec_format = methodize(load, self)
 
         if sentences is not None:
             if isinstance(sentences, GeneratorType):
@@ -1869,3 +1873,33 @@ def __iter__(self):
         model.accuracy(args.accuracy)
 
     logger.info("finished running %s", program)
+
+def methodize(func, instance):
+    return MethodType(func, instance, instance.__class__)
+
+
+def load(self, *args, **kwargs):
+    logger.warn('Load was called on instance. Calling on class instead')
+    Word2Vec.load(*args, **kwargs)
+
+
+def load_word2vec_format(
+    self,
+    fname,
+    fvocab=None,
+    binary=False,
+    encoding='utf8',
+    unicode_errors='strict',
+    limit=None,
+    datatype=REAL,
+    ):
+    logger.warn('Load was called on instance. Calling on class instead')
+    Word2Vec.load_word2vec_format(
+        fname,
+        fvocab=None,
+        binary=False,
+        encoding='utf8',
+        unicode_errors='strict',
+        limit=None,
+        datatype=REAL,
+        )