code: def plot_metrics_by_step(data, metric_name, x_label, y_label, colors):
      plt.figure(figsize=(14, 6))
  
      plt.subplot(1, 2, 1)
      plot_single_metric_by_step(
          data,
          f"train_step_{metric_name}",
          x_label,
          y_label,
          f"Train Step {metric_name.capitalize()}",
          colors[0],
      )
      plt.subplot(1, 2, 2)
      plot_single_metric_by_step(
          data,
          f"val_step_{metric_name}",
          x_label,
          y_label,
          f"Validation Step {metric_name.capitalize()}",
          colors[1],
      )
      plt.tight_layout()
  
file_path: llama-recipes/examples/plot_metrics.py
imports: import json
  import matplotlib.pyplot
  import argparse
  import os
label_desc: Use image node_id 3 for calling a global function with example usage: plot_metrics_by_step(data, metric_name, x_label, y_label, colors) without return types