code: def main(_):
      output_dir = mnist_dir("binarized_mnist")
      test_utils.remake_dir(output_dir)
      write_image_file(
          os.path.join(output_dir, _TRAIN_DATA_FILENAME), 10
      )
      write_image_file(
          os.path.join(output_dir, _VALID_DATA_FILENAME), 2
      )
      write_image_file(os.path.join(output_dir, _TEST_DATA_FILENAME), 2)
  
file_path: datasets/tensorflow_datasets/testing/fake_data_generation/binarized_mnist.py
imports: import os
  from absl import app
  from absl import flags
  import numpy
  from tensorflow_datasets.core import utils
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  from tensorflow_datasets.testing import test_utils
label_desc: Use image node_id 5 for calling a global function with example usage: main(_) without return types