From e73b89eaa36702ebbd6fb0b0d818b22d0b928964 Mon Sep 17 00:00:00 2001 From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch> Date: Thu, 29 Aug 2019 10:35:11 +0200 Subject: [PATCH] flake8 ignore --- flatland/envs/rail_env.py | 3 +-- flatland/envs/rail_generators.py | 6 ++++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index a61ef022..62efbdc5 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -271,8 +271,7 @@ class RailEnv(Environment): agent.malfunction_data['next_malfunction'] -= 1 # Only agents that have a positive rate for malfunctions and are not currently broken are considered - if agent.malfunction_data['malfunction_rate'] > 0 >= agent.malfunction_data[ - 'malfunction']: + if agent.malfunction_data['malfunction_rate'] > 0 >= agent.malfunction_data['malfunction']: # If counter has come to zero --> Agent has malfunction # set next malfunction time and duration of current malfunction diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py index 8f2c5231..40ec2e0d 100644 --- a/flatland/envs/rail_generators.py +++ b/flatland/envs/rail_generators.py @@ -683,8 +683,10 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2 0, width - 1) tries = 0 - while (station_x, station_y) in train_stations or (station_x, station_y) == node_positions[ - trainstation_node] or rail_array[(station_x, station_y)] != 0: + while (station_x, station_y) in train_stations \ + or (station_x, station_y) == node_positions[trainstation_node] \ + or rail_array[(station_x, station_y)] != 0: # noqa: E125 + station_x = np.clip( node_positions[trainstation_node][0] + np.random.randint(-node_radius, node_radius), 0, -- GitLab