fix for the high RAM usage

HUBioDataLab / DrugGEN

Official implementation of DrugGEN

GNU General Public License v3.0

48 stars 15 forks source link

Closed osbm closed 1 year ago

osbm commented 1 year ago

Added some docstrings.

osbm commented 1 year ago

If this looks ok I can reformat this file using black and remove leading white spaces.