From 23cc67b5cac09c5c7ce06cef904d31daf41f6b56 Mon Sep 17 00:00:00 2001
From: Siddhant Chaudhary <urssidd@gmail.com>
Date: Thu, 15 Aug 2024 22:53:19 +0530
Subject: [PATCH] Adding some type checks.

---
 src/indexing/codecs/residual.jl | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/indexing/codecs/residual.jl b/src/indexing/codecs/residual.jl
index 654d3e1..1ead085 100644
--- a/src/indexing/codecs/residual.jl
+++ b/src/indexing/codecs/residual.jl
@@ -48,6 +48,11 @@ function load_codec(index_path::String)
     bucket_cutoffs = JLD2.load_object(bucket_cutoffs_path)
     bucket_weights = JLD2.load_object(bucket_weights_path)
 
+    @assert centroids isa Matrix{Float32}
+    @assert avg_residual isa Float32
+    @assert bucket_cutoffs isa Vector{Float32}
+    @assert bucket_weights isa Vector{Float32}
+
     Dict(
         "centroids" => centroids,
         "avg_residual" => avg_residual,