code: def _generate_data_mass(output_dir, number_of_mammograms):
      """Generate train/test CSV and images of mass abnormalities."""
      mass_shapes = tfds.features.ClassLabel(
          names_file=tfds.core.tfds_path(
              os.path.join("image", "cbis_ddsm_mass_shapes.txt")
          )
      ).names
      mass_margins = tfds.features.ClassLabel(
          names_file=tfds.core.tfds_path(
              os.path.join("image", "cbis_ddsm_mass_margins.txt")
          )
      ).names
  
      _generate_csv(
          os.path.join(
              output_dir, "mass_case_description_train_set.csv"
          ),
          lambda: _yield_csv_rows_mass(
              output_dir, mass_shapes, mass_margins
          ),
          number_of_mammograms[0],
      )
      _generate_csv(
          os.path.join(
              output_dir, "mass_case_description_test_set.csv"
          ),
          lambda: _yield_csv_rows_mass(
              output_dir, mass_shapes, mass_margins
          ),
          number_of_mammograms[1],
      )
  
file_path: datasets/tensorflow_datasets/testing/fake_data_generation/curated_breast_imaging_ddsm.py
imports: import csv
  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
  import tensorflow_datasets.public_api
label_desc: Use image node_id 9 for calling a global function with example usage: _generate_data_mass(output_dir, number_of_mammograms) without return types