Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 8, 2024
1 parent f41df5b commit 000c1c8
Showing 1 changed file with 1 addition and 35 deletions.
36 changes: 1 addition & 35 deletions source/tests/universal/pt/model/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 000c1c8

Please sign in to comment.