fix typo

microsoft · Dec 18, 2024 · 57ab426 · 57ab426
1 parent 7791085
commit 57ab426
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eureka_ml_insights/data_utils/aime_utils.py b/eureka_ml_insights/data_utils/aime_utils.py
@@ -12,7 +12,7 @@ class AIMEExtractAnswer(DFTransformBase):
     model_answer_column: str
 
     def transform(self, df: pd.DataFrame) -> pd.DataFrame:
-        df[self.model_answer_column] = df[self.model_output_column].apply(self.parse_output_answerparse_output_answer)
+        df[self.model_answer_column] = df[self.model_output_column].apply(self.parse_output_answer)
         return df
 
     @staticmethod