Skip to content
Snippets Groups Projects
Commit 4e747784 authored by Christian Eichenberger's avatar Christian Eichenberger :badminton:
Browse files

Merge branch '52-code-cleanup-apidoc' into 'master'

52 code cleanup apidoc first steps

See merge request flatland/flatland!49
parents c2639d5c 11c0228a
No related branches found
No related tags found
No related merge requests found
"""Run benchmarks on complex rail flatland."""
import random import random
import numpy as np import numpy as np
...@@ -7,7 +8,8 @@ from flatland.envs.generators import complex_rail_generator ...@@ -7,7 +8,8 @@ from flatland.envs.generators import complex_rail_generator
from flatland.envs.rail_env import RailEnv from flatland.envs.rail_env import RailEnv
def main(): def run_benchmark():
"""Run benchmark on a small number of agents in complex rail environment."""
random.seed(1) random.seed(1)
np.random.seed(1) np.random.seed(1)
...@@ -66,4 +68,4 @@ if __name__ == "__main__": ...@@ -66,4 +68,4 @@ if __name__ == "__main__":
with Benchmarker(cycle=20, extra=1) as bench: with Benchmarker(cycle=20, extra=1) as bench:
@bench("Everything") @bench("Everything")
def _(bm): def _(bm):
main() run_benchmark()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment