modAL.models .fit() now returns self for all objects

Author: cosmic-cortex

modAL/models.py

@@ -155,8 +155,7 @@ def fit(self, X, y, bootstrap=False, **fit_kwargs):
         """
         self.X_training = X
         self.y_training = y
-        self._fit_to_known(bootstrap=bootstrap, **fit_kwargs)
-        return self
+        return self._fit_to_known(bootstrap=bootstrap, **fit_kwargs)
 
     def predict(self, X, **predict_kwargs):
         """
@@ -619,6 +618,8 @@ def fit(self, X, y, **fit_kwargs):
         for learner in self.learner_list:
             learner.fit(X, y, **fit_kwargs)
 
+        return self
+
     def query(self, X, **query_kwargs):
         """
         Finds the n_instances most informative point in the data provided by calling