diff --git a/flatland/utils/editor.py b/flatland/utils/editor.py index 7c9e0687b15f6d599ad8338906b082fc9bac7a8c..1f4a604770575f5bd24d62ade3092cbd251081be 100644 --- a/flatland/utils/editor.py +++ b/flatland/utils/editor.py @@ -174,10 +174,9 @@ class View(object): self.oRT = rt.RenderTool(self.editor.env, gl=self.sGL) def redraw(self): - # TODO: bit of a hack - can we suppress the console messages from MPL at source? - # with redirect_stdout(stdout_dest): with self.wOutput: # plt.figure(figsize=(10, 10)) + self.oRT.set_new_rail() self.oRT.renderEnv(spacing=False, arrows=False, sRailColor="gray", show=False, iSelectedAgent=self.model.iSelectedAgent, show_observations=self.show_observations()) diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index 15b5774c0e0198706ef5a959c3b4543deda04a4b..aa8731b1135326586117be2bd8e8bf6da97ea146 100644 --- a/flatland/utils/rendertools.py +++ b/flatland/utils/rendertools.py @@ -758,7 +758,7 @@ class RenderTool(object): for iAgent, agent in enumerate(self.env.agents_static): if agent is None: continue - dTargets[agent.target] = iAgent + dTargets[tuple(agent.target)] = iAgent # Draw each cell independently for r in range(env.height):