trainer.prepare_training_data() does not turn 3rd column of triplets into indices about ragatouille

[tm17-abcgen]

Can also confirm that it is working. Thanks for that fix. 👯‍♂️

[bclavie]

Hey, thank you for flagging this. It's an oversight from how the code worked in testing, where the mapping occurred at an earlier stage (when training JaColBERT). Should be fixed in #21, testing the third notebook locally with it again and will then merge + push to PyPi.

Thanks also for digging into the code to figure out the issue! This is much appreciated 😄

[bclavie]

Just finished running the notebook - haven't started training (on the move and no GPU access at the moment), but can confirm the exported triplets are now in the right format:

[0,67,40]
[0,67,19]
[0,67,28]
[0,67,83]
[0,67,17]
...

The fixed version will be on PyPi within ~10 minutes.