Commit 94e07ef9 authored by MasterScrat's avatar MasterScrat

Formatting

parent db0257ef
......@@ -25,7 +25,9 @@ You may need to install/update bazel: [Ubuntu guide](https://docs.bazel.build/ve
```
pip install ray[rllib]
pip install tensorflow # or tensorflow-gpu
pip install tensorflow
# with GPU support
# conda install -c anaconda tensorflow-gpu
pip install -r requirements.txt
```
......
......@@ -39,10 +39,16 @@ class FlatlandSparse(MultiAgentEnv):
schedule_generator = sparse_schedule_generator({float(k): float(v)
for k, v in self._config['speed_ratio_map'].items()})
env = RailEnv(width=self._config['width'], height=self._config['height'], rail_generator=rail_generator,
schedule_generator=schedule_generator, number_of_agents=self._config['number_of_agents'],
env = RailEnv(
width=self._config['width'],
height=self._config['height'],
rail_generator=rail_generator,
schedule_generator=schedule_generator,
number_of_agents=self._config['number_of_agents'],
malfunction_generator_and_process_data=malfunction_from_params(stochastic_data),
obs_builder_object=self._observation.builder(), remove_agents_at_target=False)
obs_builder_object=self._observation.builder(),
remove_agents_at_target=False
)
return env
def step(self, action_dict):
......
Markdown is supported
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