├── .env ├── .gitattribute ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE.txt ├── Pipfile ├── Pipfile.lock ├── README.md ├── data ├── 9.11.accs_by_project.csv ├── 9.12.accs_by_project.csv ├── 9.14.accs_by_project.csv ├── TypeT5-Workflow.png ├── code │ ├── bad_code_1.py │ ├── bad_code_2.py │ ├── code_with_slash.py │ ├── dummy │ │ ├── __init__.py │ │ ├── dummy_1.py │ │ └── dummy_2.py │ ├── env_code_1.py │ ├── env_code_2.py │ └── good_code_1.py ├── ex_repo │ ├── ex_code_1.py │ └── ex_code_2.py ├── mypy-dependents-by-stars.json ├── repos_split.pkl └── useful_repos.pkl ├── requirements.txt ├── scripts ├── analyze_dataset.ipynb ├── analyze_decoding_results.ipynb ├── archive │ ├── analyze_dagger.ipynb │ ├── code-t5-workflow.ipynb │ ├── debug_mypy.ipynb │ ├── fine_tune_t5.ipynb │ ├── inference_spot.ipynb │ ├── kill_dmypy.sh │ ├── test_inf_env.ipynb │ ├── train_dagger.py │ └── train_spot.ipynb ├── collect_dataset.ipynb ├── experiments │ ├── eval_file_model.ipynb │ ├── eval_func_model.py │ ├── run_hityper.ipynb │ ├── run_type4py.ipynb │ ├── run_typilus.ipynb │ └── type_check_decoding.ipynb ├── prepare_dataset.ipynb ├── run_func_decoding.ipynb ├── run_typet5.ipynb ├── scratch.ipynb └── train_model.py ├── setup.py ├── src └── typet5 │ ├── __init__.py │ ├── data.py │ ├── decode.py │ ├── experiments │ ├── __init__.py │ ├── hityper.py │ ├── type4py.py │ ├── typet5.py │ ├── typilus.py │ └── utils.py │ ├── function_dataset.py │ ├── function_decoding.py │ ├── model.py │ ├── static_analysis.py │ ├── tokenized_src.py │ ├── train.py │ ├── type_check.py │ ├── type_env.py │ ├── utils.py │ └── visualization.py └── tests ├── __init__.py ├── test_func_decoding.py ├── test_model_creation.py ├── test_static_analysis.py └── test_type_env.py /.env: -------------------------------------------------------------------------------- 1 | LIBCST_PARSER_TYPE=native 2 | -------------------------------------------------------------------------------- /.gitattribute: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-vendored 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/Pipfile -------------------------------------------------------------------------------- /Pipfile.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/Pipfile.lock -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/README.md -------------------------------------------------------------------------------- /data/9.11.accs_by_project.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/9.11.accs_by_project.csv -------------------------------------------------------------------------------- /data/9.12.accs_by_project.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/9.12.accs_by_project.csv -------------------------------------------------------------------------------- /data/9.14.accs_by_project.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/9.14.accs_by_project.csv -------------------------------------------------------------------------------- /data/TypeT5-Workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/TypeT5-Workflow.png -------------------------------------------------------------------------------- /data/code/bad_code_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/bad_code_1.py -------------------------------------------------------------------------------- /data/code/bad_code_2.py: -------------------------------------------------------------------------------- 1 | from bad_code_1 import fib 2 | 3 | i: int = 4 4 | fib(i) 5 | -------------------------------------------------------------------------------- /data/code/code_with_slash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/code_with_slash.py -------------------------------------------------------------------------------- /data/code/dummy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/code/dummy/dummy_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/dummy/dummy_1.py -------------------------------------------------------------------------------- /data/code/dummy/dummy_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/dummy/dummy_2.py -------------------------------------------------------------------------------- /data/code/env_code_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/env_code_1.py -------------------------------------------------------------------------------- /data/code/env_code_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/env_code_2.py -------------------------------------------------------------------------------- /data/code/good_code_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/code/good_code_1.py -------------------------------------------------------------------------------- /data/ex_repo/ex_code_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/ex_repo/ex_code_1.py -------------------------------------------------------------------------------- /data/ex_repo/ex_code_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/ex_repo/ex_code_2.py -------------------------------------------------------------------------------- /data/mypy-dependents-by-stars.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/mypy-dependents-by-stars.json -------------------------------------------------------------------------------- /data/repos_split.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/repos_split.pkl -------------------------------------------------------------------------------- /data/useful_repos.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/data/useful_repos.pkl -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/analyze_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/analyze_dataset.ipynb -------------------------------------------------------------------------------- /scripts/analyze_decoding_results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/analyze_decoding_results.ipynb -------------------------------------------------------------------------------- /scripts/archive/analyze_dagger.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/analyze_dagger.ipynb -------------------------------------------------------------------------------- /scripts/archive/code-t5-workflow.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/code-t5-workflow.ipynb -------------------------------------------------------------------------------- /scripts/archive/debug_mypy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/debug_mypy.ipynb -------------------------------------------------------------------------------- /scripts/archive/fine_tune_t5.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/fine_tune_t5.ipynb -------------------------------------------------------------------------------- /scripts/archive/inference_spot.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/inference_spot.ipynb -------------------------------------------------------------------------------- /scripts/archive/kill_dmypy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/kill_dmypy.sh -------------------------------------------------------------------------------- /scripts/archive/test_inf_env.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/test_inf_env.ipynb -------------------------------------------------------------------------------- /scripts/archive/train_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/train_dagger.py -------------------------------------------------------------------------------- /scripts/archive/train_spot.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/archive/train_spot.ipynb -------------------------------------------------------------------------------- /scripts/collect_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/collect_dataset.ipynb -------------------------------------------------------------------------------- /scripts/experiments/eval_file_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/eval_file_model.ipynb -------------------------------------------------------------------------------- /scripts/experiments/eval_func_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/eval_func_model.py -------------------------------------------------------------------------------- /scripts/experiments/run_hityper.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/run_hityper.ipynb -------------------------------------------------------------------------------- /scripts/experiments/run_type4py.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/run_type4py.ipynb -------------------------------------------------------------------------------- /scripts/experiments/run_typilus.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/run_typilus.ipynb -------------------------------------------------------------------------------- /scripts/experiments/type_check_decoding.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/experiments/type_check_decoding.ipynb -------------------------------------------------------------------------------- /scripts/prepare_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/prepare_dataset.ipynb -------------------------------------------------------------------------------- /scripts/run_func_decoding.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/run_func_decoding.ipynb -------------------------------------------------------------------------------- /scripts/run_typet5.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/run_typet5.ipynb -------------------------------------------------------------------------------- /scripts/scratch.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/scratch.ipynb -------------------------------------------------------------------------------- /scripts/train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/scripts/train_model.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/setup.py -------------------------------------------------------------------------------- /src/typet5/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/__init__.py -------------------------------------------------------------------------------- /src/typet5/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/data.py -------------------------------------------------------------------------------- /src/typet5/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/decode.py -------------------------------------------------------------------------------- /src/typet5/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/typet5/experiments/hityper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/experiments/hityper.py -------------------------------------------------------------------------------- /src/typet5/experiments/type4py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/experiments/type4py.py -------------------------------------------------------------------------------- /src/typet5/experiments/typet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/experiments/typet5.py -------------------------------------------------------------------------------- /src/typet5/experiments/typilus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/experiments/typilus.py -------------------------------------------------------------------------------- /src/typet5/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/experiments/utils.py -------------------------------------------------------------------------------- /src/typet5/function_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/function_dataset.py -------------------------------------------------------------------------------- /src/typet5/function_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/function_decoding.py -------------------------------------------------------------------------------- /src/typet5/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/model.py -------------------------------------------------------------------------------- /src/typet5/static_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/static_analysis.py -------------------------------------------------------------------------------- /src/typet5/tokenized_src.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/tokenized_src.py -------------------------------------------------------------------------------- /src/typet5/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/train.py -------------------------------------------------------------------------------- /src/typet5/type_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/type_check.py -------------------------------------------------------------------------------- /src/typet5/type_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/type_env.py -------------------------------------------------------------------------------- /src/typet5/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/utils.py -------------------------------------------------------------------------------- /src/typet5/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/src/typet5/visualization.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_func_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/tests/test_func_decoding.py -------------------------------------------------------------------------------- /tests/test_model_creation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/tests/test_model_creation.py -------------------------------------------------------------------------------- /tests/test_static_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/tests/test_static_analysis.py -------------------------------------------------------------------------------- /tests/test_type_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/utopia-group/TypeT5/HEAD/tests/test_type_env.py --------------------------------------------------------------------------------