From b3c17f803f0b28a43f14a9873104b9ef37c6a5aa Mon Sep 17 00:00:00 2001
From: Cedric <cedric00ik@gmail.com>
Date: Tue, 12 May 2026 09:54:55 +0000
Subject: [PATCH] temp magic

---
 scripts/evaluate_checkpoints.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/evaluate_checkpoints.py b/scripts/evaluate_checkpoints.py
index 82475a1..e1dd273 100644
--- a/scripts/evaluate_checkpoints.py
+++ b/scripts/evaluate_checkpoints.py
@@ -103,6 +103,8 @@ def main(dict_cfg: DictConfig) -> None:
     obs_processor = create_obs_processor(
         bounds_dict=training.obs_bounds.to_bounds_dict(),
         padding_masks=padding_masks,
+        num_arms=5,
+        needed_copies=5,
     )
 
     env = BrittleStarJaxEnvWrapper(