Skip to content

Commit

Permalink
black linting
Browse files Browse the repository at this point in the history
  • Loading branch information
DomInvivo committed Jan 5, 2024
1 parent bd9a178 commit 0e6c932
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions graphium/nn/architectures/global_architectures.py
Original file line number Diff line number Diff line change
Expand Up @@ -1340,6 +1340,7 @@ def _recursive_divide_dim(x: collections.abc.Mapping):
x[k] = round(v / divide_factor)
elif k in ["embed_dim"]:
x[k] = round(v / divide_factor)

_recursive_divide_dim(kwargs["layer_kwargs"])

return kwargs
Expand Down

0 comments on commit 0e6c932

Please sign in to comment.