Adding formatting.

JuliaGenAI · Sep 8, 2024 · 1bda1a8 · 1bda1a8
1 parent a631f2a
commit 1bda1a8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/examples/indexing.jl b/examples/indexing.jl
@@ -21,6 +21,7 @@ indexer = Indexer(config)
 # then big example
 config = ColBERTConfig(
     use_gpu = true,
+    checkpoint = "/home/codetalker7/models/colbertv2.0/",
     collection = "./downloads/lotte/lifestyle/dev/collection.tsv",
     index_path = "./lotte_lifestyle_index/"
 )

diff --git a/src/savers.jl b/src/savers.jl
@@ -123,7 +123,7 @@ function save_chunk_metadata_property(
         index_path::String, property::String, properties::Vector{T}) where {T}
     plan_metadata = JSON.parsefile(joinpath(index_path, "plan.json"))
     @assert plan_metadata["num_chunks"] == length(properties)
-    for chunk_idx in 1:length(properties) 
+    for chunk_idx in 1:length(properties)
         chunk_metadata = JSON.parsefile(joinpath(
             index_path, "$(chunk_idx).metadata.json"))
         chunk_metadata[property] = properties[chunk_idx]

diff --git a/test/utils.jl b/test/utils.jl
@@ -138,7 +138,7 @@ end
 @testset "kmeans_gpu_onehot!" begin
     # Test 1: When all points are centroids
     data = rand(Float32, rand(1:100), rand(1:100))
-    centroids = similar(data) 
+    centroids = similar(data)
     point_bsize = rand(1:size(data, 2))
     cluster_ids = kmeans_gpu_onehot!(data, centroids, size(data, 2))
     @test isequal(centroids[:, cluster_ids], data)