From ce2e149dbe32061e55a5aac12edc37704edaaef0 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 1 Sep 2019 09:13:20 -0400
Subject: [PATCH] updated example for flatland 2.0
---
examples/flatland_2_0_example.py | 3 ---
1 file changed, 3 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index f8a61377..a1ad9a85 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -1,5 +1,3 @@
-import time
-
import numpy as np
from flatland.envs.observations import TreeObsForRailEnv
@@ -114,7 +112,6 @@ for step in range(500):
next_obs, all_rewards, done, _ = env.step(action_dict)
env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
frame_step += 1
- time.sleep(10.1)
# Update replay buffer and train agent
for a in range(env.get_num_agents()):
agent.step((obs[a], action_dict[a], all_rewards[a], next_obs[a], done[a]))
--
GitLab