fixes a bug: parallel append! and index! failed on small datasets, n …

…< parallel_minimum_first_block
sadit · Mar 11, 2022 · d064323 · d064323 · sadit · Mar 11, 2022
1 parent 39179aa
commit d064323
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SimilaritySearch"
 uuid = "053f045d-5466-53fd-b400-a066f88fe02a"
 authors = ["Eric S. Tellez <donsadit@gmail.com>"]
-version = "0.8.10"
+version = "0.8.11"
 
 [deps]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
@@ -20,4 +20,4 @@ Intersections = "0.2.5"
 Parameters = "0.12"
 SearchModels = "0.3"
 StatsBase = "0.30.0, 0.32.0, 0.33"
-julia = "1.5"
+julia = "1.6"
diff --git a/src/graph/graph.jl b/src/graph/graph.jl
@@ -184,6 +184,8 @@ function Base.append!(
     parallel_block == 1 && return _sequential_append_loop!(index, callbacks, pools)
 
     m = 0
+
+    parallel_minimum_first_block = min(parallel_minimum_first_block, n)
     while length(index) < parallel_minimum_first_block
         m += 1
         push_item!(index, db[m], false, callbacks, pools)
@@ -216,6 +218,8 @@ function index!(
     m = 0
     db = index.db
     n = length(db)
+
+    parallel_minimum_first_block = min(parallel_minimum_first_block, n)
     while length(index) < parallel_minimum_first_block
         m += 1
         push_item!(index, db[m], false, callbacks, pools)