diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index fc426f809dec659949156c6cf7854f892e6048d2..771f228ff379ba0cd929edf9fcfb48606217c84d 100644 --- a/flatland/utils/rendertools.py +++ b/flatland/utils/rendertools.py @@ -100,7 +100,7 @@ class RenderTool(object): """ # TODO: suggest we provide an accessor in RailEnv - tbTrans = self.env.get_transitions((rcPos[0], rcPos[1], iDir)) + tbTrans = self.env.rail.get_transitions((rcPos[0], rcPos[1], iDir)) giTrans = np.where(tbTrans)[0] # RC list of transitions # HACK: workaround dead-end transitions