code: def test_custom_max_steps(self):
      env = suite_gym.load("CartPole-v1", max_episode_steps=5)
      self.assertIsInstance(env, py_environment.PyEnvironment)
      self.assertIsInstance(env, wrappers.TimeLimit)
      self.assertEqual(5, env._duration)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  from absl.testing.absltest import mock
  import gin
  import gym
  from tf_agents.environments import py_environment
  from tf_agents.environments import suite_gym
  from tf_agents.environments import wrappers
  from tf_agents.utils import test_utils
label_desc: Use image node_id 5 for calling the SuiteGymTest obj's underlying member method code with example usage: obj.test_custom_max_steps() without return types