class_docstr: Benchmark return tests for DQN Mnih15.
docstr: Benchmarks for DQN Mnih15.
  
  Args:
    output_dir: directory where to output e.g. log files.
    root_data_dir: directory under which to look for data.
    **kwargs: arbitrary named arguments. This is needed to make the
      constructor forward compatible in case PerfZero provides more named
      arguments before updating the constructor.
code: def __init__(self, output_dir=None, root_data_dir=None, **kwargs):
      """Benchmarks for DQN Mnih15.
  
      Args:
        output_dir: directory where to output e.g. log files.
        root_data_dir: directory under which to look for data.
        **kwargs: arbitrary named arguments. This is needed to make the
          constructor forward compatible in case PerfZero provides more named
          arguments before updating the constructor.
      """
      super(DqnMnih15Return, self).__init__(output_dir=output_dir)
  
file_path: agents/tf_agents/benchmark/dqn_benchmark.py
imports: import os
  import time
  import tensorflow
  from tf_agents.benchmark import utils
  from tf_agents.benchmark.perfzero_benchmark import PerfZeroBenchmark
  from tf_agents.examples.dqn.mnih15 import dqn_train_eval_atari
label_desc: Use image node_id 1 to create a new DqnMnih15Return object from inherited base classes: PerfZeroBenchmark with example: obj = DqnMnih15Return(output_dir, root_data_dir)