(self, args)
| 129 | |
| 130 | class CodeGen: |
| 131 | def __init__(self, args): |
| 132 | self.args = args |
| 133 | self.contents = {} # dictionary: maps task_id to contents (mix of tests and codes) |
| 134 | self.code_scores = {} # dictionary: maps task_id to scores (mix of tests and codes) |
| 135 | self.tests = [] |
| 136 | self.codes = [] |
| 137 | self.pair_scores = {} # dictionary: maps task_id to (code_id, test_id, score) |
| 138 | self.total_codes_requested = 0 |
| 139 | self.total_tests_requested = 0 |
| 140 | |
| 141 | logging.info("Setting up VM...") |
| 142 | |
| 143 | self.docker_execute = DockerExecute(sources_dirname=args.sources_dirname) |
| 144 | |
| 145 | logging.info("Starting LLM workers...") |
| 146 | |
| 147 | self.manager = JobManager(args) |
| 148 | |
| 149 | def add_code_or_test_job(self): |
| 150 | test_count = len(self.tests) |
nothing calls this directly
no test coverage detected