File tree Expand file tree Collapse file tree
source/tests/pt_expt/utils Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -194,18 +194,12 @@ def test_convert_regular_pt_dpa1_preserves_dense_semantics(tmp_path) -> None:
194194 from deepmd .infer import (
195195 DeepPot ,
196196 )
197- from deepmd .pt .utils .serialization import (
198- deserialize_to_file as deserialize_to_pt ,
199- )
200- from deepmd .pt .utils .serialization import (
201- serialize_from_file as serialize_from_pt ,
202- )
197+ from deepmd .pt .utils .serialization import deserialize_to_file as deserialize_to_pt
198+ from deepmd .pt .utils .serialization import serialize_from_file as serialize_from_pt
203199
204200 data = _build_dpa1_data ()
205201 descriptor_variables = data ["model" ]["descriptor" ]["@variables" ]
206- descriptor_variables ["davg" ] = np .full_like (
207- descriptor_variables ["davg" ], 0.01
208- )
202+ descriptor_variables ["davg" ] = np .full_like (descriptor_variables ["davg" ], 0.01 )
209203 source_model = tmp_path / "model.pth"
210204 converted_model = tmp_path / "model.pt2"
211205 deserialize_to_pt (str (source_model ), copy .deepcopy (data ))
You can’t perform that action at this time.
0 commit comments