Skip to content

Commit c86fe9d

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

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

deepmd/pt_expt/train/training.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1171,15 +1171,19 @@ def _compile_model(self, compile_opts: dict[str, Any]) -> None:
11711171
try:
11721172
base_ids = frozenset(
11731173
id(p)
1174-
for p in wrapper_mod.model[group_keys[0]].get_descriptor().parameters()
1174+
for p in wrapper_mod.model[group_keys[0]]
1175+
.get_descriptor()
1176+
.parameters()
11751177
)
11761178
except AttributeError:
11771179
continue
11781180
for other_key in group_keys[1:]:
11791181
try:
11801182
other_ids = frozenset(
11811183
id(p)
1182-
for p in wrapper_mod.model[other_key].get_descriptor().parameters()
1184+
for p in wrapper_mod.model[other_key]
1185+
.get_descriptor()
1186+
.parameters()
11831187
)
11841188
except AttributeError:
11851189
continue

0 commit comments

Comments
 (0)