Remove unwanted test for envpool

OpenRL-Lab · Dec 7, 2023 · d6d48af · d6d48af
1 parent 777c026
commit d6d48af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 81 deletions.
diff --git a/examples/envpool/test_model.py b/examples/envpool/test_model.py
diff --git a/examples/envpool/train_ppo.py b/examples/envpool/train_ppo.py
@@ -16,7 +16,6 @@
 
 """"""
 import numpy as np
-from test_model import evaluation
 
 from openrl.configs.config import create_config_parser
 from openrl.envs.common import make
@@ -33,7 +32,7 @@ def train():
 
     # create environment, set environment parallelism to 9
     env = make(
-        "envpool:Adventure-v5",
+        "envpool:CartPole-v1",
         render_mode=None,
         env_num=9,
         asynchronous=False,
@@ -46,7 +45,7 @@ def train():
         cfg=cfg,
     )
     # initialize the trainer
-    agent = Agent(net, use_wandb=False, project_name="envpool:Adventure-v5")
+    agent = Agent(net, use_wandb=False, project_name="envpool:CartPole-v1")
     # start training, set total number of training steps to 20000
     agent.train(total_time_steps=20000)