From 0e49156bbe35fa56082349e5c6fd0a0b09147d99 Mon Sep 17 00:00:00 2001
From: "Xingjun.Wang" <wangxingjun778@163.com>
Date: Wed, 6 Aug 2025 18:50:25 +0800
Subject: [PATCH] fix TestTinynasDamoyoloTrainerSingleGPU (#1449)

---
 tests/trainers/test_tinynas_damoyolo_trainer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/trainers/test_tinynas_damoyolo_trainer.py b/tests/trainers/test_tinynas_damoyolo_trainer.py
index 808f0a8a..52311d83 100644
--- a/tests/trainers/test_tinynas_damoyolo_trainer.py
+++ b/tests/trainers/test_tinynas_damoyolo_trainer.py
@@ -2,6 +2,7 @@
 
 import os
 import shutil
+import subprocess
 import unittest
 
 from modelscope.hub.snapshot_download import snapshot_download
@@ -19,12 +20,14 @@ def _setup():
 class TestTinynasDamoyoloTrainerSingleGPU(unittest.TestCase):
 
     def setUp(self):
+        # pycocotools==2.0.8
+        subprocess.getstatusoutput('pip install pycocotools==2.0.8')
         self.model_id = 'damo/cv_tinynas_object-detection_damoyolo'
         self.cache_path = _setup()
 
     def tearDown(self) -> None:
         super().tearDown()
-        shutil.rmtree('./workdirs')
+        shutil.rmtree('./workdirs', ignore_errors=True)
 
     @unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
     def test_trainer_from_scratch_singleGPU(self):