diff options
Diffstat (limited to 'worker')
-rw-r--r-- | worker/JOJWorker.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/worker/JOJWorker.py b/worker/JOJWorker.py index df761f8..4e17956 100644 --- a/worker/JOJWorker.py +++ b/worker/JOJWorker.py @@ -106,11 +106,16 @@ class JOJWorker(): res = {} hwNum = self.args.hw for key, value in self.hgroups.items(): - with multiprocessing.Pool(len(jojInfo["problemInfo"])) as p: - scores = p.starmap( - self.checkGroupJOJProcess, - [[HGROUP_NO, hwNum, jojInfo, fns, problemID] - for fns, problemID, _ in jojInfo["problemInfo"]]) + # TODO: re-enable multiprocessing + # with multiprocessing.Pool(len(jojInfo["problemInfo"])) as p: + # scores = p.starmap( + # self.checkGroupJOJProcess, + # [[HGROUP_NO, hwNum, jojInfo, fns, problemID] + # for fns, problemID, _ in jojInfo["problemInfo"]]) + scores = [ + self.checkGroupJOJProcess(HGROUP_NO, hwNum, jojInfo, fns, problemID) + for fns, problemID, _ in jojInfo["problemInfo"] + ] scores = [(scores[i], jojInfo["problemInfo"][i][2]) for i in range(len(scores))] self.logger.info(f"{key} h{hwNum} score {scores.__repr__()}") |