drewnutt
commented
4 months ago - Cleaned up the getitem and collatefn to no longer include forloops
- Now returns the number of atoms in the coordinates (not backwards compatible)
- Added the ability to randomly rotate and translate the retrieved example