def output_transform_tl(process_output):
    """
    Output transform for traffic light status metrics.
    """
    y_pred = process_output[0]['traffic_light_status'].argmax(dim=1)
    y = process_output[1]['traffic_light_status'].argmax(dim=1)
    return dict(y_pred=y_pred, y=y)