try to add in compute class info to task results

jeffkinnison · Dec 1, 2018 · 4547333 · 4547333
1 parent 77f57ba
commit 4547333
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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)