code: def _make_dataset(data):
      return dict(
          observed_values=data,
          observation_mask=~np.isnan(data),
          observation_index=0,
          innovation_scale=np.array(0.1, np.float32),
          observation_scale=np.array(1.0, np.float32),
          step_size=np.array(0.02, np.float32),
      )
  
imports: import functools
  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 lorenz_system
label_desc: Use image node_id 1 for calling a global function with example usage: _make_dataset(data) and returns: dict