code: def get_atari_ds_name_from_replay(
      path: str, tfds_dataset: str = "rlu_atari_checkpoints_ordered"
  ) -> str:
      """Returns `game_name` and `run_number` using replay directory `path`."""
      # `path` is assumed to be `.../game_name/run_number/replay_suffix`.
      replay_dir, _ = os.path.split(path)
      replay_dir, run_number = os.path.split(replay_dir)
      run_number = int(run_number)
      _, game = os.path.split(replay_dir)
      assert (
          run_number >= 1 and run_number <= 5
      ), "Run number must be in [1, 5]"
      return f"{tfds_dataset}/{game}_run_{run_number}"
  
imports: import os
  from absl import logging
  from dopamine.labs.offline_rl.rlu_tfds import scaling_dataset_utils
  from dopamine.labs.offline_rl.rlu_tfds import tfds_atari_utils
  import gin
label_desc: Use image node_id 1 for calling a global function with example usage: get_atari_ds_name_from_replay(path, tfds_dataset) and returns: tfds_dataset, str, game, str, run_number