1
Fork 0

feat(plotting): updated plots to not contain placeholder squares + updated colors

This commit is contained in:
Robin Meersman 2026-05-16 15:07:15 +02:00
parent fe7aecb62a
commit 708b061577
3 changed files with 35 additions and 30 deletions

View file

@ -44,6 +44,7 @@ class Columns(str, Enum):
# ... (rest of the file remains same, just need to update plotting functions and obtain_data)
"""Column names expected in every evaluation CSV."""
CHECKPOINT = "checkpoint"
ARCH = "architecture"
TIMESTEPS = "trained_timesteps"
REWARD = "eval_return"