cleanup ci

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):

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)