From 77f25d6b3a3906690a3f0b7b7bc880ab2f84f5d5 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 30 Oct 2019 15:09:29 -0400
Subject: [PATCH] fixed missing agent_malfunction_rates=None
---
flatland/envs/schedule_generators.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py
index 5883794f..9fc74368 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -199,12 +199,12 @@ def random_schedule_generator(speed_ratio_map: Optional[Mapping[float, float]] =
valid_positions.append((r, c))
if len(valid_positions) == 0:
return Schedule(agent_positions=[], agent_directions=[],
- agent_targets=[], agent_speeds=[])
+ agent_targets=[], agent_speeds=[], agent_malfunction_rates=None)
if len(valid_positions) < num_agents:
warnings.warn("schedule_generators: len(valid_positions) < num_agents")
return Schedule(agent_positions=[], agent_directions=[],
- agent_targets=[], agent_speeds=[])
+ agent_targets=[], agent_speeds=[], agent_malfunction_rates=None)
agents_position_idx = [i for i in np.random.choice(len(valid_positions), num_agents, replace=False)]
agents_position = [valid_positions[agents_position_idx[i]] for i in range(num_agents)]
--
GitLab