fix(nyz): fix drex unittest compatibility bug

opendilab · Nov 7, 2022 · ce2e0f4 · ce2e0f4
1 parent 7a3e570
commit ce2e0f4
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/dizoo/classic_control/cartpole/config/cartpole_drex_dqn_config.py b/dizoo/classic_control/cartpole/config/cartpole_drex_dqn_config.py
@@ -79,11 +79,7 @@
     ),
     env_manager=dict(type='subprocess'),
     policy=dict(type='dqn'),
+    collector=dict(type='episode'),
 )
 cartpole_drex_dqn_create_config = EasyDict(cartpole_drex_dqn_create_config)
 create_config = cartpole_drex_dqn_create_config
-
-if __name__ == "__main__":
-    # or you can enter `ding -m serial -c cartpole_drex_dqn_config.py -s 0`
-    from ding.entry import serial_pipeline
-    serial_pipeline((main_config, create_config), seed=0)