fix bug for test, remove wrong comment

opendilab · May 22, 2023 · 919c01b · 919c01b
1 parent cf50148
commit 919c01b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/ding/entry/__init__.py b/ding/entry/__init__.py
@@ -16,7 +16,6 @@
 from .application_entry import eval, collect_demo_data, collect_episodic_demo_data, \
       episode_to_transitions, episode_to_transitions_filter
 from .application_entry_trex_collect_data import trex_collecting_data, collect_episodic_demo_data_for_trex
-from .serial_entry_guided_cost import serial_pipeline_guided_cost
 from .utils import random_collect
 from .application_entry_drex_collect_data import drex_collecting_data
 from .serial_entry_mbrl import serial_pipeline_dyna, serial_pipeline_dream

diff --git a/ding/entry/tests/test_application_entry_drex_collect_data.py b/ding/entry/tests/test_application_entry_drex_collect_data.py
@@ -70,5 +70,5 @@ def test_drex_collecting_data():
     args.cfg[0].bc_iteration = 1000  # for unittest
     args.cfg[1].policy.type = 'bc'
     drex_collecting_data(args=args)
-    #os.popen('rm -rf {}'.format(expert_policy_state_dict_path))
-    #os.popen('rm -rf {}'.format(args.cfg[0].reward_model.offline_data_path))
+    os.popen('rm -rf {}'.format(expert_policy_state_dict_path))
+    os.popen('rm -rf {}'.format(args.cfg[0].reward_model.offline_data_path))