File tree Expand file tree Collapse file tree
source/tests/pt_expt/model Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments