preparing datasets for ESM3