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

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

diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 68b0c40..073c8df 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -5,7 +5,7 @@ from collections import deque
 # make sure the root path is in system path
 from pathlib import Path
 
-from flatland.envs.malfunction_generators import malfunction_from_params
+from flatland.envs.malfunction_generators import malfunction_from_params, MalfunctionParameters
 
 base_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(base_dir))
-- 
GitLab