code: def plot_prediction(ax, x, y, xnew, loc, scale, color, label):
      for k in (1, 2):
          lb = (loc - k * scale).squeeze()
          ub = (loc + k * scale).squeeze()
          alpha = 1 - 0.4 * k
          ax.fill_between(xnew, lb, ub, color=color, alpha=alpha)
  
      ax.plot(xnew, loc, color=color, label=label)
      ax.scatter(x, y, color="gray", s=8, alpha=0.7)
  
label_desc: Use image node_id 2 for calling a global function with example usage: plot_prediction(ax, x, y, xnew, loc, scale, color, label) without return types