diff --git a/source/tests/pt/test_jit.py b/source/tests/pt/test_jit.py
index f833a20a2e..d9d88ddbc1 100644
--- a/source/tests/pt/test_jit.py
+++ b/source/tests/pt/test_jit.py
@@ -9,7 +9,9 @@
 from pathlib import (
     Path,
 )
+
 import torch
+
 from deepmd.pt.entrypoints.main import (
     get_trainer,
 )
@@ -26,7 +28,6 @@
 
 
 class JITTest:
-
     def test_jit(self):
         trainer = get_trainer(deepcopy(self.config))
         trainer.run()
diff --git a/source/tests/pt/test_smooth.py b/source/tests/pt/test_smooth.py
index bacb40e87f..2e3bf61d10 100644
--- a/source/tests/pt/test_smooth.py
+++ b/source/tests/pt/test_smooth.py
@@ -26,7 +26,6 @@
 
 
 class SmoothTest:
-
     def test(
         self,
     ):