From 95da50cd13c591aab29533fd9f3b21e7c644edd9 Mon Sep 17 00:00:00 2001
From: Siddhant Chaudhary <urssidd@gmail.com>
Date: Wed, 29 May 2024 21:48:47 +0530
Subject: [PATCH] Adding `Logging` to deps, and printing info while loading a
 collection.

---
 Project.toml           | 1 +
 src/ColBERT.jl         | 1 +
 src/data/collection.jl | 1 +
 3 files changed, 3 insertions(+)

diff --git a/Project.toml b/Project.toml
index e87ed3a..65afa5a 100644
--- a/Project.toml
+++ b/Project.toml
@@ -5,6 +5,7 @@ version = "0.1.0"
 
 [deps]
 CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
+Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 Transformers = "21ca0261-441d-5938-ace7-c90938fde4d4"
 
 [compat]
diff --git a/src/ColBERT.jl b/src/ColBERT.jl
index 0da6c3b..41fb017 100644
--- a/src/ColBERT.jl
+++ b/src/ColBERT.jl
@@ -1,5 +1,6 @@
 module ColBERT
 using CSV
+using Logging
 using Transformers
 
 
diff --git a/src/data/collection.jl b/src/data/collection.jl
index 51ee22e..2b11eda 100644
--- a/src/data/collection.jl
+++ b/src/data/collection.jl
@@ -7,5 +7,6 @@ end
 
 function Collection(path::String)
     file = CSV.File(path; delim='\t', header = [:pid, :text], types = Dict(:pid => Int, :text => String), debug=true, quoted=false)
+    @info "Loaded $(length(file.text)[1]) passages."
     Collection(path, file.text)
 end