From d747195c5733c4f68d4bfbe62632d6fc1b605712 Mon Sep 17 00:00:00 2001
From: Alexander Soare <alexander.soare159@gmail.com>
Date: Sat, 4 May 2024 15:21:10 +0100
Subject: [PATCH] Fix loops in eval.py  (#128)

---
 lerobot/scripts/eval.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py
index 707bbf2e..e3afac41 100644
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -324,7 +324,7 @@ def eval_policy(
                 }
 
         # Maybe render video for visualization.
-        if max_episodes_rendered > 0:
+        if max_episodes_rendered > 0 and len(ep_frames) > 0:
             batch_stacked_frames = np.stack(ep_frames, axis=1)  # (b, t, *)
             for stacked_frames, done_index in zip(
                 batch_stacked_frames, done_indices.flatten().tolist(), strict=False
@@ -349,7 +349,6 @@ def eval_policy(
         progbar.set_postfix(
             {"running_success_rate": f"{np.mean(all_successes[:n_episodes]).item() * 100:.1f}%"}
         )
-        progbar.update()
 
     # Wait till all video rendering threads are done.
     for thread in threads: