From 68e40637bb8a229a15c0be120b5675a85ca128e7 Mon Sep 17 00:00:00 2001 From: satya131113 Date: Wed, 19 Feb 2025 03:43:42 -0700 Subject: [PATCH] Remove redundant condition in evaluate method --- langfair/metrics/stereotype/stereotype.py | 20 ++++++-------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/langfair/metrics/stereotype/stereotype.py b/langfair/metrics/stereotype/stereotype.py index 131ef1b9..d5e00dda 100644 --- a/langfair/metrics/stereotype/stereotype.py +++ b/langfair/metrics/stereotype/stereotype.py @@ -86,20 +86,12 @@ def evaluate( metric_values = {} for metric in self.metrics: if metric.name in ["Stereotype Classifier"]: - if return_data: - tmp_value = metric.evaluate( - responses=responses, - prompts=prompts, - return_data=return_data, - categories=categories, - ) - else: - tmp_value = metric.evaluate( - responses=responses, - prompts=prompts, - return_data=return_data, - categories=categories, - ) + tmp_value = metric.evaluate( + responses=responses, + prompts=prompts, + return_data=return_data, + categories=categories, + ) metric_values.update(tmp_value["metrics"]) else: metric_values[metric.name] = metric.evaluate(responses=responses)