From 151b76b77780fc6cdb45f88e810942fa3ea5bcb6 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Mon, 20 May 2019 09:10:27 +0200
Subject: [PATCH] #26 pydeps
---
examples/temporary_example.py | 11 ++++-------
tests/test_rendertools.py | 8 ++------
2 files changed, 6 insertions(+), 13 deletions(-)
diff --git a/examples/temporary_example.py b/examples/temporary_example.py
index 1f3504f2..a8194d4d 100644
--- a/examples/temporary_example.py
+++ b/examples/temporary_example.py
@@ -1,10 +1,6 @@
import random
-import numpy as np
-import matplotlib.pyplot as plt
from flatland.envs.rail_env import *
-from flatland.envs.generators import *
-from flatland.envs.observations import TreeObsForRailEnv
from flatland.utils.rendertools import *
random.seed(0)
@@ -94,7 +90,7 @@ env = RailEnv(width=7,
# print(env.obs_builder.distance_map[0, :, :, i])
# Print the observation vector for agent 0
-obs, all_rewards, done, _ = env.step({0:0})
+obs, all_rewards, done, _ = env.step({0: 0})
for i in range(env.get_num_agents()):
env.obs_builder.util_print_obs_subtree(tree=obs[i], num_features_per_node=5)
@@ -113,6 +109,7 @@ for step in range(100):
while i < len(cmds):
if cmds[i] == 'q':
import sys
+
sys.exit()
elif cmds[i] == 's':
obs, all_rewards, done, _ = env.step(action_dict)
@@ -120,9 +117,9 @@ for step in range(100):
print("Rewards: ", all_rewards, " [done=", done, "]")
else:
agent_id = int(cmds[i])
- action = int(cmds[i+1])
+ action = int(cmds[i + 1])
action_dict[agent_id] = action
- i = i+1
+ i = i + 1
i += 1
env_renderer.renderEnv(show=True)
diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py
index 245f2f32..c7841df5 100644
--- a/tests/test_rendertools.py
+++ b/tests/test_rendertools.py
@@ -4,18 +4,14 @@
Tests for `flatland` package.
"""
-from flatland.envs.rail_env import RailEnv, random_rail_generator
-import numpy as np
-#<<<<<<< HEAD
-#=======
-# import os
-#>>>>>>> dc2fa1ee0244b15c76d89ab768c5e1bbd2716147
import sys
import matplotlib.pyplot as plt
+import numpy as np
import flatland.utils.rendertools as rt
from flatland.envs.observations import TreeObsForRailEnv
+from flatland.envs.rail_env import RailEnv, random_rail_generator
def checkFrozenImage(oRT, sFileImage, resave=False):
--
GitLab