From b2eab862ab52a023a50e66df4bf8038715d34c90 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Tue, 4 Jun 2019 12:39:50 +0200
Subject: [PATCH] cleanup ci
---
flatland/utils/graphics_layer.py | 4 ++--
flatland/utils/rendertools.py | 7 +++----
2 files changed, 5 insertions(+), 6 deletions(-)
diff --git a/flatland/utils/graphics_layer.py b/flatland/utils/graphics_layer.py
index ec4641ff..d8d6d0d7 100644
--- a/flatland/utils/graphics_layer.py
+++ b/flatland/utils/graphics_layer.py
@@ -68,7 +68,7 @@ class GraphicsLayer(object):
def get_cmap(self, *args, **kwargs):
return plt.get_cmap(*args, **kwargs)
- def setRailAt(self, row, col, binTrans, iTarget=None,isSelected=False):
+ def setRailAt(self, row, col, binTrans, iTarget=None, isSelected=False):
""" Set the rail at cell (row, col) to have transitions binTrans.
The target argument can contain the index of the agent to indicate
that agent's target is at that cell, so that a station can be
@@ -76,7 +76,7 @@ class GraphicsLayer(object):
"""
pass
- def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut,isSelected=False):
+ def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut, isSelected=False):
pass
def resize(self, env):
diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py
index d7dca5d7..52892478 100644
--- a/flatland/utils/rendertools.py
+++ b/flatland/utils/rendertools.py
@@ -762,8 +762,7 @@ class RenderTool(object):
if agent is None:
continue
dTargets[tuple(agent.target)] = iAgent
- dSelected[tuple(agent.target)] = (iAgent==iSelectedAgent)
-
+ dSelected[tuple(agent.target)] = (iAgent == iSelectedAgent)
# Draw each cell independently
for r in range(env.height):
@@ -777,7 +776,7 @@ class RenderTool(object):
target = None
isSelected = False
- self.gl.setRailAt(r, c, binTrans, iTarget=target,isSelected=isSelected)
+ self.gl.setRailAt(r, c, binTrans, iTarget=target, isSelected=isSelected)
for iAgent, agent in enumerate(self.env.agents):
@@ -795,7 +794,7 @@ class RenderTool(object):
# setAgentAt uses the agent index for the color
# cmap = self.gl.get_cmap('hsv', lut=max(len(self.env.agents), len(self.env.agents_static) + 1))
- self.gl.setAgentAt(iAgent, *position, old_direction, direction, iSelectedAgent==iAgent) # ,color=cmap(iAgent))
+ self.gl.setAgentAt(iAgent, *position, old_direction, direction, iSelectedAgent == iAgent)
if show_observations:
self.renderObs(range(env.get_num_agents()), env.dev_obs_dict)
--
GitLab