code: def file_format_from_suffix(file_suffix: str) -> FileFormat:
      """Returns the file format associated with the file extension (`tfrecord`)."""
      if file_suffix not in _FILE_SUFFIX_TO_FORMAT:
          raise ValueError(
              "Unrecognized file extension: Should be one of "
              f"{list(_FILE_SUFFIX_TO_FORMAT.values())}"
          )
      return _FILE_SUFFIX_TO_FORMAT[file_suffix]
  
imports: from __future__ import annotations
  import abc
  from collections.abc import Iterator
  import enum
  import itertools
  import os
  from typing import Any, ClassVar, Dict, Iterable, List, Optional, Type, TypeVar, Union
  import uuid
  from etils import epath
  from tensorflow_datasets.core.utils import type_utils
  from tensorflow_datasets.core.utils.lazy_imports_utils import array_record_module
  from tensorflow_datasets.core.utils.lazy_imports_utils import pyarrow
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: file_format_from_suffix(file_suffix) and returns: _FILE_SUFFIX_TO_FORMAT