From e4c6ff134f3a677224532579169f498291570de6 Mon Sep 17 00:00:00 2001 From: Haibin <1400012807@pku.edu.cn> Date: Thu, 19 Dec 2024 16:28:49 +0800 Subject: [PATCH] meta tags aggregator --- tests/ops/Aggregator/test_meta_tags_aggregator.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/ops/Aggregator/test_meta_tags_aggregator.py b/tests/ops/Aggregator/test_meta_tags_aggregator.py index 75d07df9c..7aba225ae 100644 --- a/tests/ops/Aggregator/test_meta_tags_aggregator.py +++ b/tests/ops/Aggregator/test_meta_tags_aggregator.py @@ -89,26 +89,26 @@ def test_tag_list(self): { Fields.meta: [ { - MetaKeys.dialog_sentiment_label: ['开心', '平静'] + MetaKeys.dialog_sentiment_labels: ['开心', '平静'] }, { - MetaKeys.dialog_sentiment_label: ['快乐', '开心', '幸福'] + MetaKeys.dialog_sentiment_labels: ['快乐', '开心', '幸福'] }, { - MetaKeys.dialog_sentiment_label: ['难过'] + MetaKeys.dialog_sentiment_labels: ['难过'] }, { - MetaKeys.dialog_sentiment_label: ['不开心', '没头脑', '不高兴'] + MetaKeys.dialog_sentiment_labels: ['不开心', '没头脑', '不高兴'] }, { - MetaKeys.dialog_sentiment_label: ['愤怒', '愤慨'] + MetaKeys.dialog_sentiment_labels: ['愤怒', '愤慨'] } ] }, ] op = MetaTagsAggregator( api_model='qwen2.5-72b-instruct', - meta_tag_key=MetaKeys.dialog_sentiment_label, + meta_tag_key=MetaKeys.dialog_sentiment_labels, target_tags=['开心', '难过', '其他'] ) self._run_helper(op, samples)