Merge pull request #128 from vsatyamuralikrishna/refactor/remove-cond…

…ition-return-data Remove redundant condition in evaluate method
cvs-health · Feb 19, 2025 · 70c7945 · 70c7945
2 parents 5eaeb02 + 68e4063
commit 70c7945
Showing 1 changed file with 6 additions and 14 deletions.
diff --git 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)