add training_args, training_output and training_duration to the train_response

Author: ignacio

deepaas/api/v2/train.py

@@ -51,6 +51,7 @@ def build_train_response(uuid, training):
 
             ret = {}
             ret["date"] = training["date"]
+            ret["args"] = training["args"]
             ret["uuid"] = uuid
 
             if training["task"].cancelled():
@@ -62,6 +63,11 @@ def build_train_response(uuid, training):
                     ret["message"] = "%s" % exc
                 else:
                     ret["status"] = "done"
+                    ret["result"] = training["task"].result()
+                    ret["result"]["duration"] = str(datetime.datetime.strptime(ret["result"]["finish_date"],
+                                                                               '%Y-%m-%d %H:%M:%S.%f')
+                                                    - datetime.datetime.strptime(ret["date"],
+                                                                                 '%Y-%m-%d %H:%M:%S.%f'))
             else:
                 ret["status"] = "running"
             return ret
@@ -78,6 +84,7 @@ async def post(self, request, args, wsk_args=None):
             self._trainings[uuid_] = {
                 "date": str(datetime.datetime.now()),
                 "task": train_task,
+                "args": args,
             }
             ret = self.build_train_response(uuid_, self._trainings[uuid_])
             return web.json_response(ret)

deepaas/model/v2/wrapper.py

@@ -18,6 +18,7 @@
 import collections
 import concurrent.futures
 import contextlib
+import datetime
 import functools
 import multiprocessing
 import multiprocessing.pool
@@ -392,7 +393,9 @@ async def apply(self, fn, *args):
         fut = loop.create_future()
 
         def _on_done(obj):
-            loop.call_soon_threadsafe(fut.set_result, obj)
+            ret = {'output': obj,
+                   'finish_date': str(datetime.datetime.now())}
+            loop.call_soon_threadsafe(fut.set_result, ret)
 
         def _on_err(err):
             loop.call_soon_threadsafe(fut.set_exception, err)