class_docstr: Runner class to run Tandem RL experiments.
class_docstr_tok: ["Runner","class","to","run","Tandem","RL","experiments","."]
code: def __init__(self, base_dir, create_agent_fn, suite="atari"):
      if suite == "atari":
          create_environment_fn = atari_lib.create_atari_environment
      elif suite == "classic":
          create_environment_fn = gym_lib.create_gym_environment
      elif suite == "minatar":
          create_environment_fn = minatar_env.create_minatar_env
      else:
          raise ValueError(f"Unknown suite: {suite}")
      super().__init__(
          base_dir,
          create_agent_fn,
          create_environment_fn=create_environment_fn,
      )
  
imports: import sys
  from absl import logging
  from dopamine.discrete_domains import atari_lib
  from dopamine.discrete_domains import gym_lib
  from dopamine.discrete_domains import iteration_statistics
  from dopamine.discrete_domains import run_experiment
  from dopamine.labs.environments.minatar import minatar_env
  from dopamine.labs.tandem_dqn import tandem_dqn_agent
  import gin
  import numpy
  import tensorflow
label_desc: Use image node_id 1 to create a new TandemRunner object from inherited base classes: run_experiment with example: obj = TandemRunner(base_dir, create_agent_fn, suite)