def get_flat_metrics(trial_id: int, metric: str) -> List:
    full_trial_metrics = trial_metrics(trial_id)
    metrics = [m for step in full_trial_metrics['steps'] for m in step['metrics']['batch_metrics']]
    return [v[metric] for v in metrics]
def get_flat_metrics(trial_id: int, metric: str) -> List:
    full_trial_metrics = trial_metrics(trial_id)
    metrics = [m for step in full_trial_metrics['steps'] for m in step['metrics']['batch_metrics']]
    return [v[metric] for v in metrics]