786960b3ac9c0b3e75a8319649d1987194844d46,flow/envs/green_wave_env.py,TrafficLightGridEnv,k_closest_to_intersection,#TrafficLightGridEnv#Any#Any#,377

Before Change

        intersection or not. Does no padding
        
        if k < 0:
            raise IndexError("k must be greater than 0")
        dists = []

        def sort_lambda(veh_id):
            return self.get_distance_to_intersection(veh_id)

After Change

                             "parameter k={}, but k should be non-negative"
                             .format(k))

        if isinstance(edges, list):
            return [k_closest_to_intersection(edge, k) for edge in edges]

        // get the ids of all the vehicles on the edge "edges" order by
        // increasing distance to intersection
        veh_ids_ordered = sorted(
            self.k.vehicle.get_ids_by_edge(edges),
            key=self.get_distance_to_intersection)

        // return the ids of the k vehicles closest to the intersection
        return veh_ids_ordered[:k]


class PO_TrafficLightGridEnv(TrafficLightGridEnv):
    Environment used to train traffic lights to regulate traffic flow

Instances