perdim trainmax CTT 2026-07-03 workspace/scripts/eval_ctt_generated_rollout.py

workspace/scripts/eval_ctt_generated_rollout.py

@@ -1024,10 +1024,11 @@ def _scale_actions_4d(
         output = output * float(scale)
     if scale_vector is not None:
         vector = np.asarray(scale_vector, dtype=np.float32).reshape(-1)
-        if vector.size != output.shape[-1]:
-            raise ValueError(
-                "execution action scale vector length must match adapted action dimension"
-            )
+        if vector.size < output.shape[-1]:
+            pad = np.ones(output.shape[-1] - vector.size, dtype=np.float32)
+            vector = np.concatenate([vector, pad], axis=0)
+        elif vector.size > output.shape[-1]:
+            vector = vector[: output.shape[-1]]
         if np.any(~np.isfinite(vector)) or np.any(vector <= 0.0):
             raise ValueError("execution action scale vector must be positive and finite")
         output = output * vector.reshape(1, 1, 1, -1)