From a8816f18ddef1fc4eb7fb27af9199704af40d0d4 Mon Sep 17 00:00:00 2001
From: ThomasX <thomas.x@outlook.com>
Date: Fri, 8 May 2026 01:23:12 +0800
Subject: [PATCH] test: clean CP HiCache controller imports

---
 test/registered/unit/managers/test_hicache_controller_cp.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/registered/unit/managers/test_hicache_controller_cp.py b/test/registered/unit/managers/test_hicache_controller_cp.py
index 34ffc5a30..be848c02d 100644
--- a/test/registered/unit/managers/test_hicache_controller_cp.py
+++ b/test/registered/unit/managers/test_hicache_controller_cp.py
@@ -1,6 +1,6 @@
 import sys
 import types
-import unittest
+from unittest import main
 from unittest.mock import patch
 
 import torch
@@ -207,4 +207,4 @@ class TestHiCacheControllerCPWrite(CustomTestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    main()