code: def get_game_for_worker(map_name, directory_id):
      """Get game for the given worker (directory) id."""
      if map_name == "v100unfriendly":
          games = ["chopper_command", "boxing", "asterix", "seaquest"]
          worker_per_game = 5
      elif map_name == "human_nice":
          games = gym_env.ATARI_GAMES_WITH_HUMAN_SCORE_NICE
          worker_per_game = 5
      else:
          raise ValueError(
              "Unknown worker to game map name: %s" % map_name
          )
      games.sort()
      game_id = (directory_id - 1) // worker_per_game
      tf.logging.info("Getting game %d from %s." % (game_id, games))
      return games[game_id]
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import datetime
  import os
  from tensor2tensor.data_generators import gym_env
  from tensor2tensor.layers import common_video
  from tensor2tensor.models.research import rl
  from tensor2tensor.rl import rl_utils
  from tensor2tensor.rl import trainer_model_based_params
  from tensor2tensor.utils import flags
  from tensor2tensor.utils import hparam
  from tensor2tensor.utils import registry
  from tensor2tensor.utils import trainer_lib
  import tensorflow.compat.v1
label_desc: Use image node_id 21 for calling a global function with example usage: get_game_for_worker(map_name, directory_id) and returns: games