Skip to content
Snippets Groups Projects
Commit 56df476e authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

fixed agent_speed by accident

parent c6c5af12
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
flatland/envs/schedule_generators.py
+ 2
− 2
View file @ 56df476e
......@@ -263,7 +263,7 @@ def random_schedule_generator(speed_ratio_map: Optional[Mapping[float, float]] =
agents_speed = speed_initialization_helper(num_agents, speed_ratio_map, seed=_runtime_seed)
return Schedule(agent_positions=agents_position, agent_directions=agents_direction,
agent_targets=agents_target, agent_speeds=None, agent_malfunction_rates=None)
agent_targets=agents_target, agent_speeds=agents_speed, agent_malfunction_rates=None)
return generator
......@@ -307,6 +307,6 @@ def schedule_from_file(filename, load_from_package=None) -> ScheduleGenerator:
else:
agents_speed = None
return Schedule(agent_positions=agents_position, agent_directions=agents_direction,
agent_targets=agents_target, agent_speeds=None, agent_malfunction_rates=None)
agent_targets=agents_target, agent_speeds=agents_speed, agent_malfunction_rates=None)
return generator
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment