From 838139bbc7bee693cfb8d11e9a29e725bbe36ccb Mon Sep 17 00:00:00 2001
From: TAMARA JERINIC <tamara.jerinic@gmail.com>
Date: Sat, 16 Apr 2022 20:37:05 +0200
Subject: Prepravka za usaglašavanje sa frontend-om.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/microservice/api/controller.py | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

(limited to 'backend/microservice/api/controller.py')

diff --git a/backend/microservice/api/controller.py b/backend/microservice/api/controller.py
index 95ceccbb..437690ee 100644
--- a/backend/microservice/api/controller.py
+++ b/backend/microservice/api/controller.py
@@ -65,19 +65,22 @@ def train():
     files = {'file': open(filepath, 'rb')}
     r=requests.post(url, files=files)
     fileId=r.text
-    predictor = Predictor()
-    predictor._id = ""
-    predictor.username = paramsModel["username"]
-    predictor.inputs = paramsExperiment["inputColumns"]
-    predictor.output = paramsExperiment["outputColumn"]
-    predictor.isPublic = False
-    predictor.accessibleByLink = False
-    predictor.dateCreated = datetime.now()
-    predictor.experimentId = paramsExperiment["_id"]
-    predictor.modelId = paramsModel["_id"]
-    predictor.h5FileId = fileId
+    predictor = Predictor(
+        _id = "",
+        username = paramsModel["username"],
+        inputs = paramsExperiment["inputColumns"],
+        output = paramsExperiment["outputColumn"],
+        isPublic = False,
+        accessibleByLink = False,
+        dateCreated = datetime.now(),
+        experimentId = paramsExperiment["_id"],
+        modelId = paramsModel["_id"],
+        h5FileId = fileId,
+        metrics=[]
+    )
 
     print(result)
+    print(predictor)
     return jsonify(result)
 
 @app.route('/predict', methods = ['POST'])
-- 
cgit v1.2.3