code: def get_events_features():
      return {
          "collisions": tfds.features.Sequence(
              {
                  "instances": tfds.features.Tensor(
                      shape=(2,), dtype=tf.uint16
                  ),
                  "frame": tf.int32,
                  "force": tf.float32,
                  "position": tfds.features.Tensor(
                      shape=(3,), dtype=tf.float32
                  ),
                  "image_position": tfds.features.Tensor(
                      shape=(2,), dtype=tf.float32
                  ),
                  "contact_normal": tfds.features.Tensor(
                      shape=(3,), dtype=tf.float32
                  ),
              }
          )
      }
  
file_path: kubric/kubric/datasets/utils.py
imports: import json
  import numpy
  import tensorflow
  import tensorflow_datasets
  from kubric import file_io
label_desc: Use image node_id 4 for calling a global function with example usage: get_events_features() and returns: dict