diff --git a/flow/visualize/time_space_diagram.py b/flow/visualize/time_space_diagram.py index d8ae7cd73..93a3d713f 100644 --- a/flow/visualize/time_space_diagram.py +++ b/flow/visualize/time_space_diagram.py @@ -360,6 +360,8 @@ def _get_abs_pos(df, params): 'bottom_to_top': intersection / 2 + inner, 'right_to_left': junction + 3 * inner, } + elif params['network'] == HighwayNetwork: + return df['x'] else: edgestarts = defaultdict(float)