Skip to content

Commit 9e6628f

Browse files
committed
fix unittest
1 parent 4fbe9bd commit 9e6628f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/trainer/trainer_test.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -878,8 +878,7 @@ def test_trainer(self):
878878
self.assertTrue(len(os.listdir(os.path.join(checkpoint_dir, "actor"))) > 0)
879879

880880
def tearDown(self):
881-
# shutil.rmtree(self.config.checkpoint_job_dir)
882-
pass
881+
shutil.rmtree(self.config.checkpoint_job_dir)
883882

884883

885884
class TestMultiModalGRPO(BaseTrainerCase):

0 commit comments

Comments
 (0)