code: def _gen_example(x):
      return (
          x,
          {
              "image": (np.ones((16, 16, 1)) * x % 255).astype(
                  np.uint8
              ),
              "label": x % 2,
              "id": x,
          },
      )
  
imports: import pathlib
  from typing import Callable
  from unittest import mock
  import apache_beam
  from etils import epath
  import numpy
  import pytest
  import tensorflow
  from tensorflow_datasets.core import dataset_builder
  from tensorflow_datasets.core import dataset_info
  from tensorflow_datasets.core import dataset_utils
  from tensorflow_datasets.core import download
  from tensorflow_datasets.core import features
  from tensorflow_datasets.core import utils
label_desc: Use image node_id 1 for calling a global function with example usage: _gen_example(x) and returns: x,  dict