class_docstr: Test the `pixel_control_rewards` op.
class_docstr_tok: ["Test","the","`","pixel_control_rewards","`","op","."]
docstr: Compute pseudo rewards from observations.
docstr_tok: ["Compute","pseudo","rewards","from","observations","."]
code: def testPixelControlRewards(self):
      """Compute pseudo rewards from observations."""
      pseudo_rewards_tf = pixel_control_ops.pixel_control_rewards(
          self._obs_tf, self._cell
      )
  
      with self.test_session() as sess:
          self.assertAllClose(
              sess.run(
                  pseudo_rewards_tf,
                  feed_dict={self._obs_tf: self._obs_np},
              ),
              self._expected_pseudo_rewards,
          )
  
file_path: trfl/trfl/pixel_control_ops_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow.compat.v1
  from trfl import pixel_control_ops
label_desc: Use image node_id 2 for calling the PixelControlRewardsTest obj's underlying member method code with example usage: obj.testPixelControlRewards() without return types