From bbcef89013df0d15ec7bfa1d6caf63b1640f2b4e Mon Sep 17 00:00:00 2001 From: Derry Xu Date: Thu, 26 Dec 2024 22:37:18 +0000 Subject: [PATCH] update documentation --- fastchat/serve/monitor/classify/category.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/fastchat/serve/monitor/classify/category.py b/fastchat/serve/monitor/classify/category.py index 767e9d70d..b60e964ba 100644 --- a/fastchat/serve/monitor/classify/category.py +++ b/fastchat/serve/monitor/classify/category.py @@ -90,7 +90,8 @@ def post_process(self, judgement, uid): uid (str): UID of the battle to be labeled Returns: - output (Dict[str, Dict[str, str]]: Keys is battle uid, value is the output associated with that battle (usually a dictionary) + output (Dict[str, Dict[str, str]]: Key is battle UID, value is the output associated with that battle (usually a dictionary) + raw_ouput (Dict[str, str]): Key is battle UID, value is the unprocessed LLM output """ pass @@ -132,7 +133,8 @@ def post_process(labels, to_label_uids): to_label_uids (List[str]): Battle UIDs corresponding to each string that was labeled Returns: - output (Dict[str, Dict[str, str]]: Key are battle uids, value is the output associated with that battle (usually a dictionary) + output (Dict[str, Dict[str, str]]: Keys are battle uids, values are the outputs associated with that battle (usually a dictionary) + raw_ouput (Dict[str, str]): Keys is battle UIDs, value is the unprocessed HF model output or None """ pass