Checked for edge case where vehicle speeds is empty (#795)

flow-project · Dec 18, 2019 · ac23278 · ac23278
1 parent 97b962a
commit ac23278
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/flow/visualize/visualizer_rllib.py b/flow/visualize/visualizer_rllib.py
@@ -204,7 +204,12 @@ def visualizer_rllib(args):
             ret = 0
         for _ in range(env_params.horizon):
             vehicles = env.unwrapped.k.vehicle
-            vel.append(np.mean(vehicles.get_speed(vehicles.get_ids())))
+            speeds = vehicles.get_speed(vehicles.get_ids())
+
+            # only include non-empty speeds
+            if speeds:
+                vel.append(np.mean(speeds))
+
             if multiagent:
                 action = {}
                 for agent_id in state.keys():