for test

modelscope · Dec 17, 2024 · 6f44ec0 · 6f44ec0
1 parent c7662cb
commit 6f44ec0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/data_juicer/ops/mapper/query_intent_detection_mapper.py b/data_juicer/ops/mapper/query_intent_detection_mapper.py
@@ -76,6 +76,7 @@ def process_batched(self, samples, rank=None):
                                       self.use_cuda())
             results = translater(queries)
             queries = [item['translation_text'] for item in results]
+        print(queries)
 
         classifier, _ = get_model(self.model_key, rank, self.use_cuda())
         results = classifier(queries)

diff --git a/tests/ops/mapper/test_query_intent_detection_mapper.py b/tests/ops/mapper/test_query_intent_detection_mapper.py
@@ -29,9 +29,9 @@ def test_default(self):
         samples = [{
             'query': '这样好吗？'
         },{
-            'query': '把那只笔递给我。'
+            'query': '站住！'
         },{
-            'query': '难道不是这样的吗？'
+            'query': '今天阳光灿烂。'
         }
         ]
         targets = [1, 0, -1]