Refactor dimension correlation calculation in annotation_vs_evaluator.py

sotopia-lab · Jan 6, 2024 · 2b0c70a · 2b0c70a
1 parent ba8620c
commit 2b0c70a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/annotation/annotation_vs_evaluator.py b/annotation/annotation_vs_evaluator.py
@@ -111,8 +111,8 @@ def get_dimension_correlation(dimension: str) -> dict[str, float]:
         human_rewards = average_human_rewards(annotation_list)
         human_rewards_list.append(
             (
-                float(human_rewards[0][1][dimension]), # type: ignore
-                float(human_rewards[1][1][dimension]), # type: ignore
+                float(human_rewards[0][1][dimension]), 
+                float(human_rewards[1][1][dimension]),
             )
         )
     dimension_scores_agent1 = [human_rewards[0] for human_rewards in human_rewards_list]