Revised API endpoints

2020-10-02 03:28:40 +02:00
parent 7650ae2369
commit 38509c5a39
8 changed files with 143 additions and 127 deletions
--- a/model_service/model/aimodel.py
+++ b/model_service/model/aimodel.py
@ -7,8 +7,9 @@ import enum


 class AIModelType(enum.Enum):
-    SVM = 1
-    CNN = 2
+    # Optimally this would be upper case (as a convention for enums) But we want this to be the same as in the url schema
+    svm = 1
+    cnn = 2


 class AIModel(db.Model):