Skip to content

Commit 000c1c8

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f41df5b commit 000c1c8

1 file changed

Lines changed: 1 addition & 35 deletions

File tree

source/tests/universal/pt/model/test_model.py

Lines changed: 1 addition & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -5,50 +5,24 @@
55

66
from deepmd.pt.model.atomic_model import (
77
DPAtomicModel,
8-
PairTabAtomicModel,
98
)
109
from deepmd.pt.model.descriptor import (
1110
DescrptDPA1,
1211
DescrptDPA2,
1312
DescrptHybrid,
14-
DescrptSeA,
15-
DescrptSeR,
16-
DescrptSeT,
17-
DescrptSeTTebd,
1813
)
1914
from deepmd.pt.model.model import (
20-
DipoleModel,
21-
DOSModel,
2215
DPLinearModel,
23-
DPZBLModel,
24-
EnergyModel,
25-
PolarModel,
26-
PropertyModel,
27-
SpinEnergyModel,
2816
)
2917
from deepmd.pt.model.task import (
30-
DipoleFittingNet,
31-
DOSFittingNet,
3218
EnergyFittingNet,
33-
PolarFittingNet,
34-
PropertyFittingNet,
35-
)
36-
from deepmd.utils.spin import (
37-
Spin,
3819
)
3920

4021
from ....consistent.common import (
4122
parameterized,
4223
)
4324
from ...common.cases.model.model import (
44-
DipoleModelTest,
45-
DosModelTest,
46-
EnerModelTest,
4725
LinearEnerModelTest,
48-
PolarModelTest,
49-
PropertyModelTest,
50-
SpinEnerModelTest,
51-
ZBLModelTest,
5226
)
5327
from ...dpmodel.descriptor.test_descriptor import (
5428
DescriptorParamDPA1,
@@ -59,25 +33,17 @@
5933
DescriptorParamHybridMixed,
6034
DescriptorParamHybridMixedTTebd,
6135
DescriptorParamSeA,
62-
DescriptorParamSeAList,
6336
DescriptorParamSeR,
64-
DescriptorParamSeRList,
6537
DescriptorParamSeT,
66-
DescriptorParamSeTList,
6738
DescriptorParamSeTTebd,
68-
DescriptorParamSeTTebdList,
6939
)
7040
from ...dpmodel.fitting.test_fitting import (
7141
FittingParamDipole,
72-
FittingParamDipoleList,
7342
FittingParamDos,
74-
FittingParamDosList,
7543
FittingParamEnergy,
7644
FittingParamEnergyList,
7745
FittingParamPolar,
78-
FittingParamPolarList,
7946
FittingParamProperty,
80-
FittingParamPropertyList,
8147
)
8248
from ...dpmodel.model.test_model import (
8349
skip_model_tests,
@@ -885,7 +851,7 @@ def setUpClass(cls):
885851
type_map=cls.expected_type_map,
886852
)
887853
cls.module = DPLinearModel(
888-
[dp_model1,dp_model2],
854+
[dp_model1, dp_model2],
889855
type_map=cls.expected_type_map,
890856
)
891857
# only test jit API once for different models

0 commit comments

Comments
 (0)