Skip to content
Snippets Groups Projects
Commit e6051fd9 authored by Erik Nygren's avatar Erik Nygren
Browse files

fixed flake8 errors

parent 2f6d2729
No related branches found
No related tags found
No related merge requests found
...@@ -105,8 +105,7 @@ class RenderTool(object): ...@@ -105,8 +105,7 @@ class RenderTool(object):
nPixHalf = nPixCell / 2 nPixHalf = nPixCell / 2
xyHalf = array([nPixHalf, -nPixHalf]) xyHalf = array([nPixHalf, -nPixHalf])
grc2xy = array([[0, -nPixCell], [nPixCell, 0]]) grc2xy = array([[0, -nPixCell], [nPixCell, 0]])
gGrid = array(np.meshgrid(np.arange(10), -np.arange(10))) * \ gGrid = array(np.meshgrid(np.arange(10), -np.arange(10))) * array([[[nPixCell]], [[nPixCell]]])
array([[[nPixCell]], [[nPixCell]]])
# xyPixHalf = xr.DataArray([nPixHalf, -nPixHalf], # xyPixHalf = xr.DataArray([nPixHalf, -nPixHalf],
# dims="xy", # dims="xy",
# coords={"xy": ["x", "y"]}) # coords={"xy": ["x", "y"]})
...@@ -400,11 +399,11 @@ class RenderTool(object): ...@@ -400,11 +399,11 @@ class RenderTool(object):
def drawTrans2( def drawTrans2(
self, self,
xyLine, xyCentre, xyLine, xyCentre,
rotation, bDeadEnd=False, rotation, bDeadEnd=False,
sColor="gray", sColor="gray",
bArrow=True, bArrow=True,
spacing=0.1): spacing=0.1):
""" """
gLine is a numpy 2d array of points, gLine is a numpy 2d array of points,
in the plotting space / coords. in the plotting space / coords.
...@@ -493,10 +492,8 @@ class RenderTool(object): ...@@ -493,10 +492,8 @@ class RenderTool(object):
def renderEnv( def renderEnv(
self, show=False, curves=True, spacing=False, self, show=False, curves=True, spacing=False,
arrows=False, agents=True, obsrender=True, sRailColor="gray", arrows=False, agents=True, obsrender=True, sRailColor="gray", frames=False, iEpisode=None, iStep=None,
frames=False, iEpisode=None, iStep=None, iSelectedAgent=None, action_dict=None):
iSelectedAgent=None,
action_dict=None):
""" """
Draw the environment using matplotlib. Draw the environment using matplotlib.
Draw into the figure if provided. Draw into the figure if provided.
...@@ -710,11 +707,9 @@ class RenderTool(object): ...@@ -710,11 +707,9 @@ class RenderTool(object):
gP0 = array([gX1, gY1, gZ1]) gP0 = array([gX1, gY1, gZ1])
def renderEnv2( def renderEnv2(
self, show=False, curves=True, spacing=False, self, show=False, curves=True, spacing=False, arrows=False, agents=True, renderobs=True, sRailColor="gray",
arrows=False, agents=True, sRailColor="gray", frames=False, iEpisode=None, iStep=None, iSelectedAgent=None,
frames=False, iEpisode=None, iStep=None, action_dict=dict()):
iSelectedAgent=None,
action_dict=dict()):
""" """
Draw the environment using matplotlib. Draw the environment using matplotlib.
Draw into the figure if provided. Draw into the figure if provided.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment