├── .gitignore ├── README.md ├── data └── rawcode_1k.jsonl ├── environment.yaml ├── figures ├── ioexample.png └── overview.png ├── processed_data ├── codeio_1k_gens.jsonl ├── codeio_1k_gens_rev.jsonl ├── codeio_1k_gens_rev_verified.jsonl ├── codeio_1k_gens_verified.jsonl ├── codeio_1k_msg.jsonl ├── codeio_1k_msg_rev.jsonl ├── codeio_demo_final.jsonl ├── rawcode_1k.jsonl ├── rawcode_1k_msg.jsonl ├── rawcode_1k_parsed.jsonl └── rawcode_1k_unified.jsonl ├── requirements.txt ├── scripts └── pipeline_check.sh └── src ├── assemble_codeio_demo.py ├── batched_api_inference.py ├── build_codeio_msg.py ├── build_codeio_rev_msg.py ├── build_transform_msg.py ├── check_io_pred_acc_mp.py ├── check_io_pred_acc_mp_inplace.py ├── codeio_utils.py ├── parse_gen_ios.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | temp/* -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/README.md -------------------------------------------------------------------------------- /data/rawcode_1k.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/data/rawcode_1k.jsonl -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/environment.yaml -------------------------------------------------------------------------------- /figures/ioexample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/figures/ioexample.png -------------------------------------------------------------------------------- /figures/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/figures/overview.png -------------------------------------------------------------------------------- /processed_data/codeio_1k_gens.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_gens.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_1k_gens_rev.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_gens_rev.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_1k_gens_rev_verified.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_gens_rev_verified.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_1k_gens_verified.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_gens_verified.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_1k_msg.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_msg.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_1k_msg_rev.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_1k_msg_rev.jsonl -------------------------------------------------------------------------------- /processed_data/codeio_demo_final.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/codeio_demo_final.jsonl -------------------------------------------------------------------------------- /processed_data/rawcode_1k.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/rawcode_1k.jsonl -------------------------------------------------------------------------------- /processed_data/rawcode_1k_msg.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/rawcode_1k_msg.jsonl -------------------------------------------------------------------------------- /processed_data/rawcode_1k_parsed.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/rawcode_1k_parsed.jsonl -------------------------------------------------------------------------------- /processed_data/rawcode_1k_unified.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/processed_data/rawcode_1k_unified.jsonl -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/pipeline_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/scripts/pipeline_check.sh -------------------------------------------------------------------------------- /src/assemble_codeio_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/assemble_codeio_demo.py -------------------------------------------------------------------------------- /src/batched_api_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/batched_api_inference.py -------------------------------------------------------------------------------- /src/build_codeio_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/build_codeio_msg.py -------------------------------------------------------------------------------- /src/build_codeio_rev_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/build_codeio_rev_msg.py -------------------------------------------------------------------------------- /src/build_transform_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/build_transform_msg.py -------------------------------------------------------------------------------- /src/check_io_pred_acc_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/check_io_pred_acc_mp.py -------------------------------------------------------------------------------- /src/check_io_pred_acc_mp_inplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/check_io_pred_acc_mp_inplace.py -------------------------------------------------------------------------------- /src/codeio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/codeio_utils.py -------------------------------------------------------------------------------- /src/parse_gen_ios.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/parse_gen_ios.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkust-nlp/CodeIO/HEAD/src/utils.py --------------------------------------------------------------------------------