Declared trees.py public interface

dbbs-lab · May 29, 2023 · 5fa5c62 · 5fa5c62
1 parent 66e9c7c
commit 5fa5c62
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/bsb/trees.py b/bsb/trees.py
@@ -26,7 +26,7 @@ def __len__(self):
         pass
 
 
-class BoxRTree(BoxTreeInterface):
+class _BoxRTree(BoxTreeInterface):
     """
     Tree for fast lookup of queries of axis aligned rhomboids using the Rtree package.
     """
@@ -37,7 +37,7 @@ def __init__(self, boxes):
             self._rtree.insert(id, box)
 
     def __len__(self):
-        return self._rtree.get_size()
+        return len(self._rtree)
 
     def query(self, boxes, unique=False):
         """
@@ -64,5 +64,8 @@ def query(self, boxes, unique=False):
 
 
 # Cheapo provider. Needs to be a class definition so that the doc reference can be found.
-class BoxTree(BoxRTree):
+class BoxTree(_BoxRTree):
     pass
+
+
+__all__ = ["BoxTreeInterface", "BoxTree"]