wip

Author: Xyene

dmoj/graders/base.py

@@ -11,6 +11,7 @@ def __init__(self, judge, problem, language, source):
         self.binary = self._generate_binary()
         self.run_pretests_only = self.problem.meta.pretests_only
         self._abort_requested = False
+        self._current_proc = None
         self._batch_counter = 0
         self._testcase_counter = 0
 

dmoj/graders/standard.py

@@ -76,7 +76,7 @@ def check_result(self, case, result):
         return check
 
     def _launch_process(self, case):
-        self.binary.launch(
+        self._current_proc = self.binary.launch(
             time=self.problem.time_limit,
             memory=self.problem.memory_limit,
             symlinks=case.config.symlinks,