Fix circleci

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Jun 20, 2023 · 7463904 · 7463904
1 parent 8b06c45
commit 7463904
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/api.rst b/docs/api.rst
@@ -84,7 +84,7 @@ provide a natural way to compute neighbors based on the splits. We provide
 an API for extracting the nearest neighbors from a tree-model. This provides
 an API-like interface similar to :class:`~sklearn.neighbors.NearestNeighbors`.
 
-.. currentmodule:: sktree.neighbors
+.. currentmodule:: sktree
 .. autosummary::
    :toctree: generated/
 

diff --git a/sktree/__init__.py b/sktree/__init__.py
@@ -36,7 +36,8 @@
     # process, as it may not be compiled yet
 else:
     try:
-        from . import _lib, tree, ensemble, experimental, neighbors
+        from . import _lib, tree, ensemble, experimental
+        from .neighbors import NearestNeighborsMetaEstimator
         from .ensemble._unsupervised_forest import (
             UnsupervisedRandomForest,
             UnsupervisedObliqueRandomForest,
@@ -59,7 +60,7 @@
         "tree",
         "experimental",
         "ensemble",
-        "neighbors",
+        "NearestNeighborsMetaEstimator",
         "ObliqueRandomForestClassifier",
         "ObliqueRandomForestRegressor",
         "PatchObliqueRandomForestClassifier",