Merge pull request #69 from MindSetLib/asupdates

Asupdates
MindSetLib · Apr 27, 2021 · 47c23a5 · 47c23a5
2 parents 1ec2782 + 553d1a5
commit 47c23a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/insolver/model_tools/model_comparison.py b/insolver/model_tools/model_comparison.py
@@ -81,7 +81,7 @@ def _calc_metrics(self, X, y, metrics=None, stats=None, predict_params=None, fea
         trivial = InsolverTrivialWrapper(agg=lambda x: x)
         trivial.fit(X, y)
         models = [trivial] + self.models
-        features = [None] + features
+        features = [None] + features if features is not None else None
         for model in models:
             p = model.predict(X if (features is None) or (features[models.index(model)] is None)
                               else X[features[models.index(model)]],

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 
 setup(name='insolver',
-      version='0.4.9',
+      version='0.4.9b',
       description='Insolver is low-code machine learning library, initially created for the insurance industry, '
                   'but can be used in any other.\n You can find a detailed overview at '
                   'https://insolver.readthedocs.io/en/latest/source/overview.html.',