Skip to content

Commit 349ad3b

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

2 files changed

Lines changed: 4 additions & 8 deletions

File tree

source/tests/pt_expt/model/test_dpa4_export.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -619,12 +619,12 @@ def test_native_spin_chg_spin_graph_freeze(tmp_path) -> None:
619619
charge_spin slot rides the ABI tail (slot 13); metadata carries the
620620
chg-spin fields the C++/DeepEval loaders key on.
621621
"""
622+
import copy
623+
622624
from deepmd.pt_expt.descriptor.dpa4 import (
623625
DescrptDPA4,
624626
)
625-
from deepmd.pt_expt.model.get_model import (
626-
get_model as pt_expt_get_model,
627-
)
627+
from deepmd.pt_expt.model.get_model import get_model as pt_expt_get_model
628628

629629
from ...dpa4_fixtures import (
630630
jitter_zero_arrays,
@@ -633,8 +633,6 @@ def test_native_spin_chg_spin_graph_freeze(tmp_path) -> None:
633633
COMBINED_CHG_SPIN_CONFIG,
634634
)
635635

636-
import copy
637-
638636
cpu = torch.device("cpu")
639637
model = pt_expt_get_model(copy.deepcopy(COMBINED_CHG_SPIN_CONFIG))
640638
ds = model.atomic_model.descriptor

source/tests/pt_expt/model/test_dpa4_native_spin.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1051,9 +1051,7 @@ def setup_method(self) -> None:
10511051
atomic.descriptor = DescrptDPA4.deserialize(
10521052
self.pt_model.atomic_model.descriptor.serialize()
10531053
)
1054-
from deepmd.pt_expt.fitting.dpa4_ener import (
1055-
SeZMEnergyFittingNet as _FT,
1056-
)
1054+
from deepmd.pt_expt.fitting.dpa4_ener import SeZMEnergyFittingNet as _FT
10571055

10581056
atomic.fitting_net = _FT.deserialize(
10591057
self.pt_model.atomic_model.fitting_net.serialize()

0 commit comments

Comments
 (0)