From fa7fc0e8998d95296045e1a8c10add0fd37a0e8b Mon Sep 17 00:00:00 2001 From: clementou Date: Mon, 4 Mar 2024 14:59:20 -0800 Subject: [PATCH] revert formatting change --- sotopia/envs/evaluators.py | 44 +++++++++++++++----------------------- 1 file changed, 17 insertions(+), 27 deletions(-) diff --git a/sotopia/envs/evaluators.py b/sotopia/envs/evaluators.py index ba832b937..bb07496fc 100644 --- a/sotopia/envs/evaluators.py +++ b/sotopia/envs/evaluators.py @@ -427,34 +427,24 @@ def unweighted_aggregate_evaluate( ): log.debug(f"[green] The conversation is terminated. {response}") return ScriptEnvironmentResponse( - terminated=( - environment_responses[0]["terminated"] - if "terminated" in environment_responses[0] - else False - ), + terminated=environment_responses[0]["terminated"] + if "terminated" in environment_responses[0] + else False, p1_rate=( - ( - ( - agent_1_responses[0]["overall_score"] - if "overall_score" in agent_1_responses[0] - else 0 - ), - agent_1_responses[0], - ) - if agent_1_responses != ({}, "") - else None - ), + agent_1_responses[0]["overall_score"] + if "overall_score" in agent_1_responses[0] + else 0, + agent_1_responses[0], + ) + if agent_1_responses != ({}, "") + else None, p2_rate=( - ( - ( - agent_2_responses[0]["overall_score"] - if "overall_score" in agent_2_responses[0] - else 0 - ), - agent_2_responses[0], - ) - if agent_2_responses != ({}, "") - else None - ), + agent_2_responses[0]["overall_score"] + if "overall_score" in agent_2_responses[0] + else 0, + agent_2_responses[0], + ) + if agent_2_responses != ({}, "") + else None, comments=comments, )