code: def _get_dataset_cache_path(
      base_dir: str, dataset_key: DatasetKey
  ) -> str:
      return os.path.join(base_dir, dataset_key.key)
  
imports: import os
  import pickle
  import re
  import sys
  from typing import Iterable, Mapping, List, Optional, Union, Tuple
  import apache_beam
  import tensorflow
  from tfx_bsl.types import tfx_namedtuple
label_desc: Use image node_id 3 for calling a global function with example usage: _get_dataset_cache_path(base_dir, dataset_key) and returns: os