Summary

Added python module to do split of samples to train and dev set stratified by entity categories. I'll add tests when back.

To use it, from the terminal:

python src/utils/stratify_train_test_split_entities.py "path/to/dataset_to_split.jsonl" "path/to/folder/where/to/save/outputs" 0.2

Checklists

This pull/merge request meets the following requirements:

[ ] code runs
[ ] developments are ethical and secure
[ ] you have made proportionate checks that the code works correctly
[ ] test suite passes
[ ] developments adhere to AQA plan (see docs/aqa/aqa_plan.md)
[ ] data log updated (see docs/aqa/data_log.md), if necessary
[ ] assumptions, and caveats log updated (see docs/aqa/assumptions_caveats.md), if necessary
[ ] minimum usable documentation written in the docs folder

Comments have been added below around the incomplete checks.