Fixed stuff

model_service/schemas/aimodel_schema.py

@@ -5,5 +5,11 @@ from model import AIModel
 
 
 class AIModelSchema(ModelSchema):
+
+    default = fields.Method("boolize_default", dump_only=True)
+
+    def boolize_default(self, ai_model) -> bool:
+        return bool(ai_model.default)
+
     class Meta:
         model = AIModel

model_service/schemas/default_schema.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
-from marshmallow import fields
-from marshmallow_sqlalchemy import ModelSchema
+from marshmallow import fields, Schema
 
 
-class DefaultSchema(ModelSchema):
+class DefaultSchema(Schema):
     id = fields.UUID()