fix

Acellera · Apr 26, 2024 · 34cf86c · 34cf86c
1 parent 298dbee
commit 34cf86c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/reinvent/reinvent.py b/scripts/reinvent/reinvent.py
@@ -315,7 +315,8 @@ def create_env_fn():
             # Add data to the replay buffer
             reward = replay_data.get(("next", "reward"))
             replay_data.set("priority", reward)
-            experience_replay_buffer.extend(replay_data)
+            if len(replay_data) > 0:
+                experience_replay_buffer.extend(replay_data)
 
         # Log info
         if logger: