Minor fix in collection functions.

JuliaGenAI · Jun 16, 2024 · d6e769f · d6e769f
1 parent 8082de9
commit d6e769f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/data/collection.jl b/src/data/collection.jl
@@ -13,11 +13,11 @@ function Collection(path::String)
 end
 
 function get_chunksize(collection::Collection, nranks::Int)
-    min(25000, 1 + floor(length(collection.data) / nranks))
+    Int(min(25000, 1 + floor(length(collection.data) / nranks)))
 end
 
 function enumerate_batches(
-        collection::Collection, chunksize::Union{Int, Missing} = missing,
+        collection::Collection; chunksize::Union{Int, Missing} = missing,
         nranks::Union{Int, Missing} = missing)
     if ismissing(chunksize)
         if ismissing(nranks)