code: def load_fake_dataset():
      """Return some fake data for testing purposes."""
      x_train = jnp.arange(0.0, 120.0) / 120.0
      y_train = x_train + jnp.sin(x_train * 10.0) + x_train * x_train
      x_train = x_train[..., jnp.newaxis]
      return x_train, y_train[..., jnp.newaxis]
  
file_path: probability/tensorflow_probability/python/experimental/autobnn/util.py
imports: from typing import Any, Callable, List, Tuple
  import jax
  import jax.numpy
  import scipy
  from tensorflow_probability.python.experimental.autobnn import bnn
  from tensorflow_probability.substrates.jax.distributions import distribution
label_desc: Use image node_id 3 for calling a global function with example usage: load_fake_dataset() and returns: x_train,  y_train