def packMapsTrainingScenarios(scenarios): out = [] for team, teamConfig in scenarios.items(): packedTeam = [] for vehClass, i in VEHICLE_CLASS_INDICES.items(): config = teamConfig.get(vehClass) if config is None: continue duration = [ config['duration']['initial'], config['duration']['max'] ] goals = [config['goals'][vClass] for vClass in VEHICLE_CLASSES] packedTeam.append([i, duration, goals]) out.append([team, packedTeam]) return out