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 Oct 27, 2023
1 parent 4b7aff0 commit 13fc1db
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion deepmd/utils/tabulate.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,10 @@ def __init__(
# functype
if activation_fn == ACTIVATION_FN_DICT["tanh"]:
self.functype = 1
elif activation_fn in (ACTIVATION_FN_DICT["gelu"], ACTIVATION_FN_DICT["gelu_tf"]):
elif activation_fn in (
ACTIVATION_FN_DICT["gelu"],
ACTIVATION_FN_DICT["gelu_tf"],
):
self.functype = 2
elif activation_fn == ACTIVATION_FN_DICT["relu"]:
self.functype = 3
Expand Down

0 comments on commit 13fc1db

Please sign in to comment.