Update multi_agent_training.py

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))