From f380125f5f4c3436f4d6f7d4bcbb982eb5fdc5da Mon Sep 17 00:00:00 2001
From: Erik Nygren <baerenjesus@gmail.com>
Date: Tue, 9 Jun 2020 09:36:50 +0000
Subject: [PATCH] Update multi_agent_inference.py
---
torch_training/multi_agent_inference.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/torch_training/multi_agent_inference.py b/torch_training/multi_agent_inference.py
index e131f46..cb8fa32 100644
--- a/torch_training/multi_agent_inference.py
+++ b/torch_training/multi_agent_inference.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.predictions import ShortestPathPredictorForRailEnv
from flatland.envs.rail_env import RailEnv
@@ -37,10 +37,11 @@ n_agents = 10
observation_builder = TreeObsForRailEnv(max_depth=2)
# Use a the malfunction generator to break agents from time to time
-stochastic_data = {'malfunction_rate': 8000, # Rate of malfunction occurence of single agent
- 'min_duration': 15, # Minimal duration of malfunction
- 'max_duration': 50 # Max duration of malfunction
- }
+stochastic_data = MalfunctionParameters(malfunction_rate=10000, # Rate of malfunction occurence
+ min_duration=15, # Minimal duration of malfunction
+ max_duration=50 # Max duration of malfunction
+ )
+
# Custom observation builder
--
GitLab