diff --git a/scripts/plot.py b/scripts/plot.py index 9199dee..81854cf 100644 --- a/scripts/plot.py +++ b/scripts/plot.py @@ -48,12 +48,14 @@ def read_metrics(filename: str) -> pd.DataFrame: line, ).groups() - data.append({ - "episode": int(episode), - "winning_rate": float(winning_rate), - "mean_score": float(mean_score), - "max_tile": int(max_tile), - }) + data.append( + { + "episode": int(episode), + "winning_rate": float(winning_rate), + "mean_score": float(mean_score), + "max_tile": int(max_tile), + } + ) return pd.DataFrame(data)