fix FIGS numpy scalar extraction warning

csinva · Aug 22, 2023 · e94d115 · e94d115
1 parent 86645d0
commit e94d115
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/tree/figs.html b/docs/tree/figs.html
@@ -499,7 +499,7 @@
 
         def _predict_tree_single_point(root: Node, x):
             if root.left is None and root.right is None:
-                return root.value
+                return root.value[0, 0]
             left = x[root.feature] &lt;= root.threshold
             if left:
                 if root.left is None:  # we don&#39;t actually have to worry about this case
@@ -1116,7 +1116,7 @@ <h2 id="params">Params</h2>
 
         def _predict_tree_single_point(root: Node, x):
             if root.left is None and root.right is None:
-                return root.value
+                return root.value[0, 0]
             left = x[root.feature] &lt;= root.threshold
             if left:
                 if root.left is None:  # we don&#39;t actually have to worry about this case

diff --git a/imodels/tree/figs.py b/imodels/tree/figs.py
@@ -476,7 +476,7 @@ def _predict_tree(self, root: Node, X):
 
         def _predict_tree_single_point(root: Node, x):
             if root.left is None and root.right is None:
-                return root.value
+                return root.value[0, 0]
             left = x[root.feature] <= root.threshold
             if left:
                 if root.left is None:  # we don't actually have to worry about this case