diff --git a/rollout.py b/rollout.py index f32b336d9644ac7d6c9ca814ae887686c3f39ae7..33c9291d02c515cb2060b43769d1dfff3d4d5290 100644 --- a/rollout.py +++ b/rollout.py @@ -20,6 +20,7 @@ def main(): episode_count = 0 while episode_count < 20: _, _, done, _ = env.step(env.action_space.sample()) + print("Completed a step") if done: episode_count += 1 print(episode_count)