Enabling dump from SupervisedModel

bigmlcom · Apr 19, 2022 · e56317b · e56317b
1 parent 2a94d18
commit e56317b
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -3,6 +3,12 @@
 History
 -------
 
+7.2.2 (2022-04-20)
+~~~~~~~~~~~~~~~~~~
+
+- Enabling dump from SupervisedModel class.
+
+
 7.2.1 (2022-04-19)
 ~~~~~~~~~~~~~~~~~~
 

diff --git a/bigml/supervised.py b/bigml/supervised.py
@@ -134,3 +134,11 @@ def predict_probability(self, *args, **kwargs):
         except TypeError:
             del new_kwargs["missing_strategy"]
             return self.local_model.predict_probability(*args, **new_kwargs)
+
+    def dump(self, **kwargs):
+        """Delegate to local model"""
+        self.local_model.dump(**kwargs)
+
+    def dumps(self):
+        """Delegate to local model"""
+        return self.local_model.dumps()
diff --git a/bigml/version.py b/bigml/version.py
@@ -1 +1 @@
-__version__ = '7.2.1'
+__version__ = '7.2.2'