Fix type instability in make_knn_heaps

dillondaudert · Jul 25, 2019 · d66457b · d66457b
1 parent 97c93b3
commit d66457b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -64,7 +64,8 @@ function make_knn_heaps(data::Vector{V},
                                           M <: SemiMetric}
     np = length(data)
     D = result_type(metric, data[1], data[1])
-    knn_heaps = [BinaryMaxHeap{NNTuple{Int, D}}() for _ in 1:np]
+    HeapType = BinaryMaxHeap{NNTuple{Int, D}}
+    knn_heaps = HeapType[HeapType() for _ in 1:np]
     for i in 1:np
         k_idxs = sample_neighbors(np, n_neighbors, exclude=[i])
         for j in k_idxs