Skip to content

Commit

Permalink
Add traceback to metadata
Browse files Browse the repository at this point in the history
  • Loading branch information
ovidner committed Jan 6, 2023
1 parent b3629e7 commit 768340c
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions src/omnsga/drivers.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import random
import traceback
import warnings
from copy import deepcopy
from itertools import chain
Expand Down Expand Up @@ -195,7 +196,7 @@ def _setup_driver(self, problem):
)
self.toolbox = toolbox

self.evaluation_metadata = {"generation": 0}
self.evaluation_metadata = {"generation": 0, "success": 1, "msg": ""}

def _get_recorder_metadata(self, case_name):
metadata = super()._get_recorder_metadata(case_name)
Expand All @@ -209,11 +210,16 @@ def evaluate_individual(self, individual):
self.set_design_var(name, value)

with RecordingDebugging(self._get_name(), self.iter_count, self):
success = True
try:
self._problem().model.run_solve_nonlinear()
except AnalysisError:
success = False
self.evaluation_metadata["success"] = 0
self.evaluation_metadata["msg"] = traceback.format_exc()
else:
success = True
self.evaluation_metadata["success"] = 1
self.evaluation_metadata["msg"] = ""

self.iter_count += 1

Expand Down

0 comments on commit 768340c

Please sign in to comment.