Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Apr 2, 2024
1 parent bd89e3a commit bd14b45
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/pt/model/task/fitting.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,7 +326,7 @@ def get_sel_type(self) -> List[int]:
def __setitem__(self, key, value):
if key in ["bias_atom_e"]:
if isinstance(value, dict):
#TODO: need to update to handle multiple properties
# TODO: need to update to handle multiple properties
value = list(value.values())[0].view([self.ntypes, self._net_out_dim()])
else:
value = value.view([self.ntypes, self._net_out_dim()])
Expand Down

0 comments on commit bd14b45

Please sign in to comment.