code: def radon_log_likelihood_fn(
      params,
      log_uranium,
      floor,
      county,
      floor_by_county,
      log_radon,
      reduce_sum=True,
  ):
      log_likelihood = make_radon_observation_dist(
          params, log_uranium, floor, county, floor_by_county
      ).log_prob(log_radon)
      if reduce_sum:
          return tf.reduce_sum(log_likelihood, [-1])
      return log_likelihood
  
imports: import functools
  import tensorflow.compat.v2
  import tensorflow_probability
  from tensorflow_probability.python.internal import dtype_util
  from inference_gym.internal import data
  from inference_gym.targets import bayesian_model
  from inference_gym.targets import model
  from inference_gym.targets.ground_truth import radon_contextual_effects_indiana
  from inference_gym.targets.ground_truth import radon_contextual_effects_indiana_halfnormal
  from inference_gym.targets.ground_truth import radon_contextual_effects_minnesota
  from inference_gym.targets.ground_truth import radon_contextual_effects_minnesota_halfnormal
label_desc: Use image node_id 4 for calling a global function with example usage: radon_log_likelihood_fn(params, log_uranium, floor, county, floor_by_county, log_radon, reduce_sum) and returns: log_likelihood, tf