search

choderalab / mtenn

Modular Training and Evaluation of Neural Networks
MIT License
5 stars 1 forks source link

Update LigandOnlyModel return to match other Model classes. #31

Closed kaminow closed 9 months ago

kaminow commented 9 months ago

Update the LigandOnlyModel forward pass to return the prediction and the prediction in a list to match the signature of the other Model classes.