docstr: Test ERWR with Cartpole-v1 environment.
docstr_tok: ["Test","ERWR","with","Cartpole-v1","environment","."]
code: def test_erwr_cartpole(self):
      """Test ERWR with Cartpole-v1 environment."""
      with TFTrainer(snapshot_config, sess=self.sess) as trainer:
          deterministic.set_seed(1)
          env = GymEnv("CartPole-v1")
  
          policy = CategoricalMLPPolicy(
              name="policy", env_spec=env.spec, hidden_sizes=(32, 32)
          )
  
          baseline = LinearFeatureBaseline(env_spec=env.spec)
  
          sampler = LocalSampler(
              agents=policy,
              envs=env,
              max_episode_length=env.spec.max_episode_length,
              is_tf_worker=True,
          )
  
          algo = ERWR(
              env_spec=env.spec,
              policy=policy,
              baseline=baseline,
              sampler=sampler,
              discount=0.99,
          )
  
          trainer.setup(algo, env)
  
          last_avg_ret = trainer.train(n_epochs=10, batch_size=10000)
          assert last_avg_ret > 60
  
          env.close()
  
file_path: garage/tests/garage/tf/algos/test_erwr.py
imports: import pytest
  from garage.envs import GymEnv
  from garage.experiment import deterministic
  from garage.np.baselines import LinearFeatureBaseline
  from garage.sampler import LocalSampler
  from garage.tf.algos import ERWR
  from garage.tf.policies import CategoricalMLPPolicy
  from garage.trainer import TFTrainer
  from tests.fixtures import snapshot_config, TfGraphTestCase
label_desc: Use image node_id 1 for calling the TestERWR obj's underlying member method code with example usage: obj.test_erwr_cartpole() without return types