From 4547333d53b6b9338ffd0a7cb5ec4c44c6efebfc Mon Sep 17 00:00:00 2001 From: Jeff Kinnison Date: Sat, 1 Dec 2018 18:06:29 -0500 Subject: [PATCH] try to add in compute class info to task results --- shadho/shadho.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/shadho/shadho.py b/shadho/shadho.py index a60d047..1cf807f 100644 --- a/shadho/shadho.py +++ b/shadho/shadho.py @@ -408,6 +408,11 @@ def success(self, tag, loss, results): """ # Get bookkeeping information from the task tag result_id, model_id, ccid = tag.split('.') + results['compute_class'] = { + 'id': ccid, + 'name': self.ccs[ccid].name, + 'value': self.ccs[ccid].value + } # Update the DB with the result self.backend.register_result(model_id, result_id, loss, results)