Method write_code

codegen/codegen.py:277–308

(self)

                self.handle_judge_pair(task_id=task_id, score=score)

    def write_code(self):
        # Delete any existing code from a previous run
        delete_old_scripts(args.sources_dirname, args.function_name)

        try:
            while True:
                self.handle_results()

                active_workers = self.manager.active_workers()
                approx_queue_depth = self.manager.approx_queue_depth()
                if approx_queue_depth == 0 and active_workers < args.workers:
                    logging.info(f"Work queue empty and detected only {active_workers}/{args.workers} workers active.  Adding job...")
                    self.add_code_or_test_job()
                else:
                    logging.info(f"Work queue depth = {approx_queue_depth} active workers = {active_workers}/{args.workers}")

                for pair_score in self.pair_scores.values():
                    (code_id, test_id, score) = pair_score
                    if score is None:
                        continue

                    if score >= args.threshold:
                        logging.info(f"Found a good code/test pair: code={code_id} test={test_id} score={score}")
                        copy_candidate_scripts(args.sources_dirname, args.function_name, code_id, test_id)
                        logging.info("Wrote final code and test to disk. Exiting...")
                        return
        except KeyboardInterrupt:
            logging.info("Terminating early on user request...")
        except Exception as e:
            logging.error(f"Exception in run_queue: {e}")
        finally:
            self.manager.terminate()

def main(args):
    logging.info(f"Input comments: {args.comments}")