Skip to content

Commit 2275dbd

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

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

source/tests/pt_expt/utils/test_graph_pt2_metadata.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff 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))

0 commit comments

Comments
 (0)