code: def get_builder(dataset, data_dir):
      return tfds.builder(dataset, data_dir=data_dir, try_gcs=True)
  
imports: import collections
  import dataclasses
  import functools
  import itertools
  from typing import Any, Callable, Dict, Iterator, Optional, Sequence, Union
  from absl import logging
  from flax.training import common_utils
  import jax
  import jax.numpy
  import numpy
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 13 for calling a global function with example usage: get_builder(dataset, data_dir) and returns: tfds