code: def get_estimator(estimator_name: str, params) -> _AutoBnnEstimator:
      return NAME_TO_ESTIMATOR[estimator_name](**params)
  
imports: from typing import Any, Mapping, Optional, Sequence
  import jax
  import jax.numpy
  from jaxtyping import ArrayLike, PyTree
  from tensorflow_probability.python.experimental.autobnn import bnn
  from tensorflow_probability.python.experimental.autobnn import likelihoods
  from tensorflow_probability.python.experimental.autobnn import models
  from tensorflow_probability.python.experimental.autobnn import training_util
label_desc: Use image node_id 1 for calling a global function with example usage: get_estimator(estimator_name, params) and returns: NAME_TO_ESTIMATOR