code: def _test_dataset():
      return dict(
          train_locations=np.arange(20).reshape((10, 2)),
          train_extents=1.0 + (np.arange(10) % 10),
          train_counts=50.0 + np.arange(10),
      )
  
file_path: probability/spinoffs/inference_gym/inference_gym/targets/log_gaussian_cox_process_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from inference_gym.internal import test_util
  from inference_gym.targets import log_gaussian_cox_process
label_desc: Use image node_id 1 for calling a global function with example usage: _test_dataset() and returns: dict