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
                  ),
              }
          )
      }
  
imports: import dataclasses
  import json
  import logging
  import imageio
  import numpy
  import png
  import tensorflow
  import tensorflow_datasets.public_api
  from typing import List, Dict, Union
label_desc: Use image node_id 4 for calling a global function with example usage: get_events_features() and returns: dict