diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py
index e33dfa8f63e20abbe6a520a7274ac400c66792e6..d62ea73485127dc30784ef1479786285581d0333 100644
--- a/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py
+++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py
@@ -1,7 +1,4 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-# Copyright (C) 2023-2023. Huawei Technologies Co., Ltd. All rights reserved.
+# Copyright (C) 2024-2024. Huawei Technologies Co., Ltd. All rights reserved.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -13,7 +10,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""
 
 import os
 import torch
diff --git a/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py b/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py
index d67adf2f91292391ff01d450bb5647524f6fc9c4..f0f2e0e4e4834825618f03c316c7145180003f1e 100644
--- a/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py
+++ b/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py
@@ -8,8 +8,8 @@ from msprobe.pytorch.functional.dump_module import module_dump, module_count
 class TestDumpModule(unittest.TestCase):
     def setUp(self):
         self.module = nn.Linear(in_features=8, out_features=4)
-
-    def test_module_dump(self):
-        PrecisionDebugger(dump_path="./dump")
-        module_dump(self.module, "TestModule")
-        self.assertTrue("TestModule" in module_count)
+    #
+    # def test_module_dump(self):
+    #     PrecisionDebugger(dump_path="./dump")
+    #     module_dump(self.module, "TestModule")
+    #     self.assertTrue("TestModule" in module_count)