code: def episode_id(episode):
      return f'{episode["checkpoint_id"]}_{episode["episode_id"]}'
  
file_path: datasets/tensorflow_datasets/rl_unplugged/atari_utils.py
imports: from __future__ import annotations
  import dataclasses
  from typing import Any, Dict
  import numpy
  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: episode_id(episode) and returns: episode, str, episode