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 Jan 11, 2024
1 parent 72f7fc6 commit e3914e8
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions deepmd_utils/model_format/output_def.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def forward(
**kwargs,
):
ret = cls.forward(self, *args, **kwargs)
keys = self.md.keys_outp() if output_type == 'model' else self.md.keys()
keys = self.md.keys_outp() if output_type == "model" else self.md.keys()
for kk in keys:
dd = self.md[kk]
check_var(ret[kk], dd)
Expand All @@ -50,13 +50,16 @@ def forward(
check_var(ret[dnr], self.md[dnr])
check_var(ret[dnc], self.md[dnc])
return ret

return wrapper


def model_check_output(cls):
return check_output(cls, 'model')
return check_output(cls, "model")


def fitting_check_output(cls):
return check_output(cls, 'fitting')
return check_output(cls, "fitting")


class VariableDef:
Expand Down

0 comments on commit e3914e8

Please sign in to comment.