class_docstr: Modified tests from dopamine run_experiment_test.py.
class_docstr_tok: ["Modified","tests","from","dopamine","run_experiment_test.py","."]
code: def _agent_step(self, rewards, observations):
      # We verify that rewards are clipped (and set by MockEnvironment as a
      # function of observation)
      # observation = observations[0]
      # expected_rewards = [1 if observation % 2 else -1]
      # self.assertEqual(expected_reward, reward)
      actions = [ob % 2 for ob in observations]
      return actions
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import shutil
  from absl import flags
  from dopamine.discrete_domains import logger
  import mock
  import numpy
  from tensor2tensor.rl import dopamine_connector
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the BatchedRunnerTest obj's underlying member method code with example usage: obj._agent_step(rewards, observations) and returns: actions