code: def _plot_noise_fn(
      noise_scale: jax.Array, ax_t: plt.Axes, ax_h: plt.Axes
  ) -> plt.Figure:
      ax_t.plot(noise_scale.squeeze())
      ax_t.set_xlim(0, noise_scale.shape[0])
      ax_t.set_ylim(0)
  
      ax_h.hist(noise_scale.squeeze(), orientation="horizontal")
      ax_h.set_ylim(0)
      ax_h.axes.get_yaxis().set_visible(False)
      ax_h.axes.get_xaxis().set_visible(False)
      return ax_t.figure
  
imports: import functools
  from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
  import bayeux
  import jax
  import jax.numpy
  from jaxtyping import PyTree
  import matplotlib.pyplot
  import numpy
  import pandas
  from tensorflow_probability.python.experimental.autobnn import bnn
  from tensorflow_probability.python.experimental.autobnn import util
  from tensorflow_probability.python.experimental.timeseries import metrics
  import tensorflow_probability.substrates.jax
label_desc: Use image node_id 9 for calling a global function with example usage: _plot_noise_fn(noise_scale, ax_t, ax_h) and returns: ax_t