data_path_prefix=/scratch/nlp/wutong/dataset/PoSE-Datasets


# python -u evaluation/make_kv_retrieval_data.py \
#     --num-keys 75 \
#     --num-examples 500 \
#     --output-path ${data_path_prefix}/kv_data/kv-retrieval-75_keys.jsonl.gz


# python -u evaluation/make_kv_retrieval_data.py \
#     --num-keys 140 \
#     --num-examples 500 \
#     --output-path ${data_path_prefix}/kv_data/kv-retrieval-140_keys.jsonl.gz


python -u evaluation/make_kv_retrieval_data.py \
    --num-keys 300 \
    --num-examples 500 \
    --output-path ${data_path_prefix}/kv_data/kv-retrieval-300_keys.jsonl.gz



python -u evaluation/make_kv_retrieval_data.py \
    --num-keys 230 \
    --num-examples 500 \
    --output-path ${data_path_prefix}/kv_data/kv-retrieval-230_keys.jsonl.gz
