fix: plots script for camelcase

lkaesberg · lkaesberg · commit 813b6df36cef · 2024-11-23T18:00:52.000+01:00
diff --git a/mallm/evaluation/plotting/plots.py b/mallm/evaluation/plotting/plots.py
@@ -17,7 +17,7 @@ def process_stats_file(file_path: str) -> pd.DataFrame:
     data = json.loads(Path(file_path).read_text())
     # Extract only the average scores
     return pd.DataFrame(
-        {k: v["average_score"] for k, v in data.items() if "average_score" in v},
+        {k: v["averageScore"] for k, v in data.items() if "averageScore" in v},
         index=[0],
     )
 

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def process_stats_file(file_path: str) -> pd.DataFrame:`
`17`	`17`	`data = json.loads(Path(file_path).read_text())`
`18`	`18`	`# Extract only the average scores`
`19`	`19`	`return pd.DataFrame(`
`20`		`- {k: v["average_score"] for k, v in data.items() if "average_score" in v},`
	`20`	`+ {k: v["averageScore"] for k, v in data.items() if "averageScore" in v},`
`21`	`21`	`index=[0],`
`22`	`22`	`)`
`23`	`23`