code: def test_gym_registration_with_kwargs(self):
      reg_id, env = gym_utils.register_gym_env(
          "tensor2tensor.rl.gym_utils_test:EnvWithOptions",
          kwargs={"done_action": 2},
      )
  
      self.assertEqual("T2TEnv-EnvWithOptions-v0", reg_id)
  
      # Obligatory reset.
      env.reset()
  
      # Make sure that on action = 0, 1 we are not done, but on 2 we are.
      _, _, done, _ = env.step(0)
      self.assertFalse(done)
  
      _, _, done, _ = env.step(1)
      self.assertFalse(done)
  
      _, _, done, _ = env.step(2)
      self.assertTrue(done)
  
      # Now lets try to change the env -- note we have to change the version.
      reg_id, env = gym_utils.register_gym_env(
          "tensor2tensor.rl.gym_utils_test:EnvWithOptions",
          version="v1",
          kwargs={"done_action": 1},
      )
  
      self.assertEqual("T2TEnv-EnvWithOptions-v1", reg_id)
  
      # Obligatory reset.
      env.reset()
  
      # Make sure that on action = 0, 2 we are not done, but on 1 we are.
      _, _, done, _ = env.step(0)
      self.assertFalse(done)
  
      _, _, done, _ = env.step(2)
      self.assertFalse(done)
  
      _, _, done, _ = env.step(1)
      self.assertTrue(done)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import unittest
  import gym
  from gym import spaces
  import numpy
  from tensor2tensor.rl import gym_utils
  import tensorflow.compat.v1
label_desc: Use image node_id 13 for calling the GymUtilsTest obj's underlying member method code with example usage: obj.test_gym_registration_with_kwargs() without return types