code: def image_feature(size: int, doc=None) -> tfds.features.Image:
      return tfds.features.Image(
          shape=(size, size, 3),
          dtype=np.uint8,
          encoding_format="png",
          doc=doc,
      )
  
imports: import dataclasses
  import enum
  from typing import Any, Dict, List, Mapping
  from etils import epath
  import h5py
  import numpy
  import tensorflow_datasets.public_api
  import tree
label_desc: Use image node_id 6 for calling a global function with example usage: image_feature(size, doc) and returns: tfds