|
5 | 5 |
|
6 | 6 | from deepmd.pt.model.atomic_model import ( |
7 | 7 | DPAtomicModel, |
8 | | - PairTabAtomicModel, |
9 | 8 | ) |
10 | 9 | from deepmd.pt.model.descriptor import ( |
11 | 10 | DescrptDPA1, |
12 | 11 | DescrptDPA2, |
13 | 12 | DescrptHybrid, |
14 | | - DescrptSeA, |
15 | | - DescrptSeR, |
16 | | - DescrptSeT, |
17 | | - DescrptSeTTebd, |
18 | 13 | ) |
19 | 14 | from deepmd.pt.model.model import ( |
20 | | - DipoleModel, |
21 | | - DOSModel, |
22 | 15 | DPLinearModel, |
23 | | - DPZBLModel, |
24 | | - EnergyModel, |
25 | | - PolarModel, |
26 | | - PropertyModel, |
27 | | - SpinEnergyModel, |
28 | 16 | ) |
29 | 17 | from deepmd.pt.model.task import ( |
30 | | - DipoleFittingNet, |
31 | | - DOSFittingNet, |
32 | 18 | EnergyFittingNet, |
33 | | - PolarFittingNet, |
34 | | - PropertyFittingNet, |
35 | | -) |
36 | | -from deepmd.utils.spin import ( |
37 | | - Spin, |
38 | 19 | ) |
39 | 20 |
|
40 | 21 | from ....consistent.common import ( |
41 | 22 | parameterized, |
42 | 23 | ) |
43 | 24 | from ...common.cases.model.model import ( |
44 | | - DipoleModelTest, |
45 | | - DosModelTest, |
46 | | - EnerModelTest, |
47 | 25 | LinearEnerModelTest, |
48 | | - PolarModelTest, |
49 | | - PropertyModelTest, |
50 | | - SpinEnerModelTest, |
51 | | - ZBLModelTest, |
52 | 26 | ) |
53 | 27 | from ...dpmodel.descriptor.test_descriptor import ( |
54 | 28 | DescriptorParamDPA1, |
|
59 | 33 | DescriptorParamHybridMixed, |
60 | 34 | DescriptorParamHybridMixedTTebd, |
61 | 35 | DescriptorParamSeA, |
62 | | - DescriptorParamSeAList, |
63 | 36 | DescriptorParamSeR, |
64 | | - DescriptorParamSeRList, |
65 | 37 | DescriptorParamSeT, |
66 | | - DescriptorParamSeTList, |
67 | 38 | DescriptorParamSeTTebd, |
68 | | - DescriptorParamSeTTebdList, |
69 | 39 | ) |
70 | 40 | from ...dpmodel.fitting.test_fitting import ( |
71 | 41 | FittingParamDipole, |
72 | | - FittingParamDipoleList, |
73 | 42 | FittingParamDos, |
74 | | - FittingParamDosList, |
75 | 43 | FittingParamEnergy, |
76 | 44 | FittingParamEnergyList, |
77 | 45 | FittingParamPolar, |
78 | | - FittingParamPolarList, |
79 | 46 | FittingParamProperty, |
80 | | - FittingParamPropertyList, |
81 | 47 | ) |
82 | 48 | from ...dpmodel.model.test_model import ( |
83 | 49 | skip_model_tests, |
@@ -885,7 +851,7 @@ def setUpClass(cls): |
885 | 851 | type_map=cls.expected_type_map, |
886 | 852 | ) |
887 | 853 | cls.module = DPLinearModel( |
888 | | - [dp_model1,dp_model2], |
| 854 | + [dp_model1, dp_model2], |
889 | 855 | type_map=cls.expected_type_map, |
890 | 856 | ) |
891 | 857 | # only test jit API once for different models |
|
0 commit comments