From cbe1aab405f7ce78f59a16632a5b780f4a884062 Mon Sep 17 00:00:00 2001 From: Aditya Desai Date: Sat, 21 Jun 2025 21:50:43 -0700 Subject: [PATCH] Fix: json dump silent failures (#75) --- vcache/vcache_policy/strategies/benchmark_verified_global.py | 4 ++-- vcache/vcache_policy/strategies/verified.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/vcache/vcache_policy/strategies/benchmark_verified_global.py b/vcache/vcache_policy/strategies/benchmark_verified_global.py index 8fa8842..a6488d7 100644 --- a/vcache/vcache_policy/strategies/benchmark_verified_global.py +++ b/vcache/vcache_policy/strategies/benchmark_verified_global.py @@ -215,9 +215,9 @@ def update_metadata( metadata: The metadata of the embedding. """ if is_correct: - self.global_observations.append((round(similarity_score, 3), 1)) + self.global_observations.append((round(similarity_score, 3).item(), 1)) else: - self.global_observations.append((round(similarity_score, 3), 0)) + self.global_observations.append((round(similarity_score, 3).item(), 0)) def select_action( self, similarity_score: float, metadata: EmbeddingMetadataObj diff --git a/vcache/vcache_policy/strategies/verified.py b/vcache/vcache_policy/strategies/verified.py index f37289c..d27901a 100644 --- a/vcache/vcache_policy/strategies/verified.py +++ b/vcache/vcache_policy/strategies/verified.py @@ -484,9 +484,9 @@ def add_observation_to_metadata( metadata: EmbeddingMetadataObj - The metadata of the embedding """ if is_correct: - metadata.observations.append((round(similarity_score, 3), 1)) + metadata.observations.append((round(similarity_score, 3).item(), 1)) else: - metadata.observations.append((round(similarity_score, 3), 0)) + metadata.observations.append((round(similarity_score, 3).item(), 0)) def select_action( self, similarity_score: float, metadata: EmbeddingMetadataObj