From feb9357bc0f11aaa21b49d88af4a90635e3296de Mon Sep 17 00:00:00 2001
From: Erik Nygren <baerenjesus@gmail.com>
Date: Tue, 9 Jun 2020 09:37:25 +0000
Subject: [PATCH] Update render_agent_behavior.py

---
 torch_training/render_agent_behavior.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py
index 82706a4..bfe5ba0 100644
--- a/torch_training/render_agent_behavior.py
+++ b/torch_training/render_agent_behavior.py
@@ -3,7 +3,7 @@ from collections import deque
 
 import numpy as np
 import torch
-from flatland.envs.malfunction_generators import malfunction_from_params
+from flatland.envs.malfunction_generators import malfunction_from_params, MalfunctionParameters
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.envs.rail_generators import sparse_rail_generator
-- 
GitLab