Merge pull request #138 from ImogenBits/solver_exceptions

Fix program run exception handling
Algorithmic-Battle · Oct 25, 2023 · 8d21543 · 8d21543
2 parents c2ee22c + fa40605
commit 8d21543
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/algobattle/battle.py b/algobattle/battle.py
@@ -175,8 +175,9 @@ async def run(
             set_cpus=self.set_cpus,
             ui=ui,
         )
-        if gen_result.instance is None:
+        if gen_result.info.error is not None:
             return Fight(score=1, max_size=max_size, generator=gen_result.info, solver=None)
+        assert gen_result.instance is not None
 
         sol_result = await self.solver.run(
             gen_result.instance,
@@ -189,8 +190,9 @@ async def run(
             set_cpus=self.set_cpus,
             ui=ui,
         )
-        if sol_result.solution is None:
+        if sol_result.info.error is not None:
             return Fight(score=0, max_size=max_size, generator=gen_result.info, solver=sol_result.info)
+        assert sol_result.solution is not None
 
         if self.problem.with_solution:
             assert gen_result.solution is not None