code: def write_task(output_dir, task_id, task):
      path = os.path.join(output_dir, "{}.json".format(task_id))
      with tf.io.gfile.GFile(path, "w") as f:
          json.dump(task, f)
  
file_path: datasets/tensorflow_datasets/testing/fake_data_generation/arc.py
imports: import json
  import os
  from absl import app
  from absl import flags
  import numpy
  from tensorflow_datasets.core.utils import tfds_write_path
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  from tensorflow_datasets.testing import test_utils
label_desc: Use image node_id 6 for calling a global function with example usage: write_task(output_dir, task_id, task) without return types