bug fix

yzhao062 · Mar 15, 2021 · 5d3066e · 5d3066e
1 parent 6b569d7
commit 5d3066e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -25,4 +25,5 @@ v<0.1.0>, <02/17/2020> -- Documentation update.
 v<0.1.0>, <02/17/2020> -- Code cleaning.
 v<0.1.1>, <06/15/2020> -- Unit test update.
 v<0.1.1>, <07/01/2020> -- Documentation update.
-v<0.1.2>, <12/22/2020> -- Refactor for new version of sklearn.
+v<0.1.2>, <12/22/2020> -- Refactor for new version of sklearn.
+v<0.1.3>, <03/14/2021> -- Bug fix for the range of aom and moa.
diff --git a/combo/models/score_comb.py b/combo/models/score_comb.py
@@ -68,7 +68,8 @@ def _aom_moa_helper(mode, scores, n_buckets, method, bootstrap_estimators,
     # use random_state instead
     # for now it is fixed at n_estimators/2
     n_estimators = scores.shape[1]
-    check_parameter(n_buckets, 2, n_estimators, param_name='n_buckets')
+    check_parameter(n_buckets, 2, n_estimators, include_left=True,
+                    include_right=True, param_name='n_buckets')
 
     scores_buckets = np.zeros([scores.shape[0], n_buckets])
 

diff --git a/combo/version.py b/combo/version.py
@@ -20,4 +20,4 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = '0.1.2'  # pragma: no cover
+__version__ = '0.1.3'  # pragma: no cover