The dummy example (truncated to 100 MP compounds) runs to completion just fine despite the OOM error with the full dataset #8. Thanks for troubleshooting getting this up and running on Google Colab. Feel free to use or modify the following script if desired - might make for a nice badge to include at the top of the README if you decide you want to use/adapt it.

The dummy example (truncated to 100 MP compounds) runs to completion just fine despite the OOM error with the full dataset #8. Thanks for troubleshooting getting this up and running on Google Colab. Feel free to use or modify the following script if desired - might make for a nice badge to include at the top of the README if you decide you want to use/adapt it.