Release v2.5.10

Fixup the KD-Tree algorithm

.github/workflows/deploy.yml

@@ -12,7 +12,7 @@ on:
 env:
   MAJOR: ${{ 2 }}
   MINOR: ${{ 5 }}
-  FIXUP: ${{ 9 }}
+  FIXUP: ${{ 10 }}
   PACKAGE_INIT_FILE: ${{ 'divik/__init__.py' }}
   PACKAGE_INIT_FILE_VERSION_LINE: ${{ 1 }}
   PACKAGE_SETUP_FILE: ${{ 'setup.py' }}

README.md

@@ -40,7 +40,7 @@ docker pull gmrukwa/divik
 To install specific version, you can specify it in the command, e.g.:
 
 ```bash
-docker pull gmrukwa/divik:2.5.9
+docker pull gmrukwa/divik:2.5.10
 ```
 
 ## Python package
@@ -79,7 +79,7 @@ pip install divik
 or any stable tagged version, e.g.:
 
 ```bash
-pip install divik==2.5.9
+pip install divik==2.5.10
 ```
 
 If you want to have compatibility with

divik/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '2.5.9'
+__version__ = '2.5.10'
 
 from ._summary import plot, reject_split
 

divik/cluster/_kmeans/_initialization.py

@@ -156,13 +156,16 @@ def make_tree(X, leaf_size: int, _feature_idx: int = 0, selector=None) \
         selector = np.ones((X.shape[0],), dtype=bool)
     if selector.sum() < 2 * leaf_size:
         centroid = X[selector, :].mean(axis=0, keepdims=True)
-        return Leaf(centroid, X.shape[0])
+        return Leaf(centroid, int(selector.sum()))
     feature = X[selector, _feature_idx]
     thr = np.mean(feature)
     left_idx = selector.copy()
     left_idx[selector] = feature < thr
     right_idx = selector.copy()
     right_idx[selector] = np.logical_not(left_idx[selector])
+    if left_idx[selector].all() or right_idx[selector].all():
+        centroid = X[selector, :].mean(axis=0, keepdims=True)
+        return Leaf(centroid, int(selector.sum()))
     next_feature = (_feature_idx + 1) % X.shape[1]
     return Node(
         left=make_tree(X, leaf_size=leaf_size,

docs/instructions/installation.rst

@@ -14,7 +14,7 @@ To install latest stable version use::
 
 To install specific version, you can specify it in the command, e.g.::
 
-    docker pull gmrukwa/divik:2.5.9
+    docker pull gmrukwa/divik:2.5.10
 
 Python package
 --------------
@@ -31,7 +31,7 @@ package::
 
 or any stable tagged version, e.g.::
 
-    pip install divik==2.5.9
+    pip install divik==2.5.10
 
 If you want to have compatibility with
 `gin-config <https://github.com/google/gin-config>`_, you can install

setup.py

@@ -6,7 +6,7 @@
 import sys
 import numpy
 
-__version__ = '2.5.9'
+__version__ = '2.5.10'
 
 LINUX_OPTS = {
     'extra_link_args': [