fix to saving extra info

clvrai · Mar 11, 2024 · 83e5aa0 · 83e5aa0
1 parent 37ac30f
commit 83e5aa0
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/sprint/saycan_eval.py b/sprint/saycan_eval.py
@@ -178,8 +178,6 @@ def multiprocess_rollout(
         average_return=np.mean(rollout_returns),
         subgoal_success=np.mean(subgoal_successes),
     )
-    for key, value in extra_info.items():
-        rollout_metrics[key] = np.mean(value)
     # make a WandB table for the high level skill, ground truth sequence, predicted, completed skills
     saycan_completed_skill_data = []
     keys = [