tests/process_data_test.py

!python scripts/process_dataset.py \
    --random_state 42 \
    --test_split_rate 0.2 \
    --augment