ENH: add __all__ dicts to kdtree and ckdtree in spatial.

This cleans up the spatial namespace quite a bit.
scipy · Mar 10, 2011 · 4fd7424 · 4fd7424
1 parent 3dcb8a4
commit 4fd7424
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 3 deletions.
diff --git a/scipy/spatial/ckdtree.c b/scipy/spatial/ckdtree.c
diff --git a/scipy/spatial/ckdtree.pyx b/scipy/spatial/ckdtree.pyx
@@ -8,7 +8,7 @@ import kdtree
 
 cdef double infinity = np.inf
 
-
+__all__ = ['cKDTree']
 
 
 # priority queue

diff --git a/scipy/spatial/kdtree.py b/scipy/spatial/kdtree.py
@@ -5,6 +5,11 @@
 from heapq import heappush, heappop
 import scipy.sparse
 
+__all__ = ['minkowski_distance_p', 'minkowski_distance',
+           'distance_matrix',
+           'Rectangle', 'KDTree']
+
+
 def minkowski_distance_p(x,y,p=2):
     """Compute the pth power of the L**p distance between x and y