style: make torch.nn imports more uniform

DeepRank / deeprank2

An open-source deep learning framework for data mining of protein-protein interfaces or single-residue variants.

https://deeprank2.readthedocs.io/en/latest/?badge=latest

Apache License 2.0

32 stars 12 forks source link

style: make torch.nn imports more uniform #626

Closed DaniBodor closed 1 month ago

DaniBodor commented 1 month ago

All functions from torch.nn.functional are now explicitly imported by name and used directly.
In all other cases where classes from torch.nn are used, we do from torch import nn and then use nn.<Class> throughout.

Note that this was also changed in one docstring in fba9e56. Not sure if that messes up the rendering of the documents. That commit can easily be dropped if needed.

fix #269

DaniBodor commented 1 month ago

Maybe just revert it for simplicity.

I reverted it.