code: def get_dataset_name(env_name: str) -> str:
      return DATASET_NAMES[env_name]
  
file_path: acme/examples/baselines/imitation/helpers.py
imports: from typing import Tuple
  from acme import wrappers
  import dm_env
  import gym
  import numpy
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: get_dataset_name(env_name) and returns: DATASET_NAMES