e673c301c42f64a5ff53b78effdf78b0855e83e5,cistar-dev/cistar/core/base_env.py,SumoEnvironment,reset,#SumoEnvironment#,317

Before Change

        self.traci_connection.simulationStep()

        // TODO: Replace these traci calls with initial_state accesses
        for veh_id in self.ids:
            self.traci_connection.vehicle.setSpeedMode(veh_id, 0)
            if veh_id in self.rl_ids:
                self.traci_connection.vehicle.setLaneChangeMode(veh_id, 0)
            self.vehicles[veh_id]["type"] = self.traci_connection.vehicle.getTypeID(veh_id)
            self.vehicles[veh_id]["edge"] = self.traci_connection.vehicle.getRoadID(veh_id)
            self.vehicles[veh_id]["position"] = self.traci_connection.vehicle.getLanePosition(veh_id)
            self.vehicles[veh_id]["lane"] = self.traci_connection.vehicle.getLaneIndex(veh_id)
            self.vehicles[veh_id]["speed"] = self.traci_connection.vehicle.getSpeed(veh_id)
            self.vehicles[veh_id]["fuel"] = self.traci_connection.vehicle.getFuelConsumption(veh_id)
            self.vehicles[veh_id]["distance"] = self.traci_connection.vehicle.getDistance(veh_id)
            self.vehicles[veh_id]["absolute_position"] = self.get_x_by_id(veh_id)


        self.state = self.getState()
        observation = np.copy(self.state)

        return observation

After Change

            // re-initialize perceived state
            self.vehicles[veh_id] = self.initial_vehicles_state[veh_id]

            print("veh id:", veh_id)
            print("edge:", self.vehicles[veh_id]["edge"])
            print("position:", self.vehicles[veh_id]["position"])
            print("lane:", self.vehicles[veh_id]["edge"])
            print("speed:", self.vehicles[veh_id]["edge"])

Instances