flashrag-dev 0.1.4.dev20241125__tar.gz → 0.1.4.dev20241203__tar.gz

{flashrag-dev-0.1.4.dev20241125/flashrag_dev.egg-info → flashrag-dev-0.1.4.dev20241203}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag-dev
-Version: 0.1.4.dev20241125
+Version: 0.1.4.dev20241203
 Summary: A library for efficient Retrieval-Augmented Generation research
 Home-page: https://github.com/RUC-NLPIR/FlashRAG
 Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou

{flashrag-dev-0.1.4.dev20241125 → flashrag-dev-0.1.4.dev20241203}/flashrag/evaluator/metrics.py

@@ -52,7 +52,7 @@ class F1_Score(BaseMetric):
     def __init__(self, config):
         super().__init__(config)
 
-    def token_level_scores(self, prediction: str, ground_truths: str):
+    def token_level_scores(self, prediction: str, ground_truths: list):
         final_metric = {"f1": 0, "precision": 0, "recall": 0}
         if isinstance(ground_truths, str):
             ground_truths = [ground_truths]

{flashrag-dev-0.1.4.dev20241125 → flashrag-dev-0.1.4.dev20241203}/flashrag/pipeline/active_pipeline.py

@@ -981,8 +981,6 @@ class IRCOTPipeline(BasicPipeline):
 
             # Perform batch retrieval for new thoughts of active items
             if active_item_ids:
-                print(len(active_item_ids))
-                print(active_item_ids)
                 new_thoughts_for_retrieval = [batch_thoughts[item_id][-1] for item_id in active_item_ids]
                 new_retrieval_results, new_scoress = self.retriever.batch_search(new_thoughts_for_retrieval, return_score=True)
 

{flashrag-dev-0.1.4.dev20241125 → flashrag-dev-0.1.4.dev20241203}/flashrag/pipeline/pipeline.py

@@ -141,8 +141,8 @@ class ConditionalPipeline(BasicPipeline):
         """
 
         super().__init__(config, prompt_template)
-        self.judger = get_judger(config)
         self.generator = get_generator(config)
+        self.judger = get_judger(config)
         self.retriever = get_retriever(config)
 
         self.sequential_pipeline = SequentialPipeline(
@@ -187,10 +187,9 @@ class AdaptivePipeline(BasicPipeline):
     ):
         super().__init__(config)
         # load adaptive classifier as judger
-        self.judger = get_judger(config)
-
         generator = get_generator(config)
         retriever = get_retriever(config)
+        self.judger = get_judger(config)
         self.generator = generator
         self.retriever = retriever
 

flashrag-dev-0.1.4.dev20241203/flashrag/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.4dev20241203"

{flashrag-dev-0.1.4.dev20241125 → flashrag-dev-0.1.4.dev20241203/flashrag_dev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag-dev
-Version: 0.1.4.dev20241125
+Version: 0.1.4.dev20241203
 Summary: A library for efficient Retrieval-Augmented Generation research
 Home-page: https://github.com/RUC-NLPIR/FlashRAG
 Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou

flashrag-dev-0.1.4.dev20241125/flashrag/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.4dev20241125"