diff --git a/source/tests/universal/pt/model/test_model.py b/source/tests/universal/pt/model/test_model.py
index 261f45d3ec..e8eb288762 100644
--- a/source/tests/universal/pt/model/test_model.py
+++ b/source/tests/universal/pt/model/test_model.py
@@ -5,50 +5,24 @@
 
 from deepmd.pt.model.atomic_model import (
     DPAtomicModel,
-    PairTabAtomicModel,
 )
 from deepmd.pt.model.descriptor import (
     DescrptDPA1,
     DescrptDPA2,
     DescrptHybrid,
-    DescrptSeA,
-    DescrptSeR,
-    DescrptSeT,
-    DescrptSeTTebd,
 )
 from deepmd.pt.model.model import (
-    DipoleModel,
-    DOSModel,
     DPLinearModel,
-    DPZBLModel,
-    EnergyModel,
-    PolarModel,
-    PropertyModel,
-    SpinEnergyModel,
 )
 from deepmd.pt.model.task import (
-    DipoleFittingNet,
-    DOSFittingNet,
     EnergyFittingNet,
-    PolarFittingNet,
-    PropertyFittingNet,
-)
-from deepmd.utils.spin import (
-    Spin,
 )
 
 from ....consistent.common import (
     parameterized,
 )
 from ...common.cases.model.model import (
-    DipoleModelTest,
-    DosModelTest,
-    EnerModelTest,
     LinearEnerModelTest,
-    PolarModelTest,
-    PropertyModelTest,
-    SpinEnerModelTest,
-    ZBLModelTest,
 )
 from ...dpmodel.descriptor.test_descriptor import (
     DescriptorParamDPA1,
@@ -59,25 +33,17 @@
     DescriptorParamHybridMixed,
     DescriptorParamHybridMixedTTebd,
     DescriptorParamSeA,
-    DescriptorParamSeAList,
     DescriptorParamSeR,
-    DescriptorParamSeRList,
     DescriptorParamSeT,
-    DescriptorParamSeTList,
     DescriptorParamSeTTebd,
-    DescriptorParamSeTTebdList,
 )
 from ...dpmodel.fitting.test_fitting import (
     FittingParamDipole,
-    FittingParamDipoleList,
     FittingParamDos,
-    FittingParamDosList,
     FittingParamEnergy,
     FittingParamEnergyList,
     FittingParamPolar,
-    FittingParamPolarList,
     FittingParamProperty,
-    FittingParamPropertyList,
 )
 from ...dpmodel.model.test_model import (
     skip_model_tests,
@@ -885,7 +851,7 @@ def setUpClass(cls):
             type_map=cls.expected_type_map,
         )
         cls.module = DPLinearModel(
-            [dp_model1,dp_model2],
+            [dp_model1, dp_model2],
             type_map=cls.expected_type_map,
         )
         # only test jit API once for different models