Merge pull request #287 from tomwhite/numba-warnings-0.4dev

Fix more NumbaWarnings
lmcinnes · Aug 29, 2019 · 0ecc2a0 · 0ecc2a0
2 parents 54d8019 + aebf623
commit 0ecc2a0
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/umap/rp_tree.py b/umap/rp_tree.py
@@ -447,7 +447,6 @@ def sparse_euclidean_random_projection_split(inds, indptr, data, indices, rng_st
     return indices_left, indices_right, hyperplane, hyperplane_offset
 
 
-@numba.jit()
 def make_euclidean_tree(data, indices, rng_state, leaf_size=30):
     if indices.shape[0] > leaf_size:
         left_indices, right_indices, hyperplane, offset = euclidean_random_projection_split(
@@ -466,7 +465,6 @@ def make_euclidean_tree(data, indices, rng_state, leaf_size=30):
     return node
 
 
-@numba.jit()
 def make_angular_tree(data, indices, rng_state, leaf_size=30):
     if indices.shape[0] > leaf_size:
         left_indices, right_indices, hyperplane, offset = angular_random_projection_split(
@@ -485,7 +483,6 @@ def make_angular_tree(data, indices, rng_state, leaf_size=30):
     return node
 
 
-@numba.jit()
 def make_sparse_euclidean_tree(inds, indptr, data, indices, rng_state, leaf_size=30):
     if indices.shape[0] > leaf_size:
         left_indices, right_indices, hyperplane, offset = sparse_euclidean_random_projection_split(
@@ -508,7 +505,6 @@ def make_sparse_euclidean_tree(inds, indptr, data, indices, rng_state, leaf_size
     return node
 
 
-@numba.jit()
 def make_sparse_angular_tree(inds, indptr, data, indices, rng_state, leaf_size=30):
     if indices.shape[0] > leaf_size:
         left_indices, right_indices, hyperplane, offset = sparse_angular_random_projection_split(

diff --git a/umap/umap_.py b/umap/umap_.py
@@ -413,7 +413,6 @@ def compute_membership_strengths(knn_indices, knn_dists, sigmas, rhos):
     return rows, cols, vals
 
 
-@numba.jit()
 def fuzzy_simplicial_set(
     X,
     n_neighbors,