def _sample_vector_dist(rng):
  """Sample a vector distribution config."""
  choice = rng.choice(["normal", "uniform"])
  if choice == "normal":
    mean = rng.normal(0, 1)
    std = rng.uniform(0, 2)
    return choice, {"mean": mean, "std": std}
  elif choice == "uniform":
    min_v = rng.uniform(-5, 2.5)
    max_v = min_v + rng.uniform(0, 5)
    return choice, {"min": min_v, "max": max_v}