fix(dspy): use DataFrame.applymap in older pandas

bakebrain · Mar 13, 2024 · e53fcbe · e53fcbe
1 parent 2d845de
commit e53fcbe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dspy/evaluate/evaluate.py b/dspy/evaluate/evaluate.py
@@ -184,7 +184,10 @@ def wrapped_program(example_idx, example):
         df = pd.DataFrame(data)
 
         # Truncate every cell in the DataFrame
-        df = df.map(truncate_cell)
+        if hasattr(df, "map"):  # DataFrame.applymap was renamed to DataFrame.map in Pandas 2.1.0
+            df = df.map(truncate_cell)
+        else:
+            df = df.applymap(truncate_cell)
 
         # Rename the 'correct' column to the name of the metric object
         assert callable(metric)