From 49bd71e3ee0602dfd4c48bfa309da0ff4a7f216e Mon Sep 17 00:00:00 2001
From: noiji <52301388+noiji@users.noreply.github.com>
Date: Fri, 15 Nov 2024 13:52:04 +0900
Subject: [PATCH] Fix wrong key

---
 experiments/llmlingua2/data_collection/filter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/experiments/llmlingua2/data_collection/filter.py b/experiments/llmlingua2/data_collection/filter.py
index 5b31417..0b703aa 100644
--- a/experiments/llmlingua2/data_collection/filter.py
+++ b/experiments/llmlingua2/data_collection/filter.py
@@ -66,7 +66,7 @@
     kept["labels"],
     kept["origin"],
     kept["comp"],
-    res_pt["retrieval"],
+    kept["retrieval"],
     kept["comp_rate"],
     kept["variation_rate"],
     kept["hitting_rate"],