Added saving target class name

2020-10-01 19:38:34 +02:00
parent bafd6be131
commit 62ea95cfe4
4 changed files with 7 additions and 5 deletions
--- a/model_service/schemas/info_schema.py
+++ b/model_service/schemas/info_schema.py
@ -8,6 +8,7 @@ import uuid

 class InfoSchema(Schema):
    id = fields.UUID(default=uuid.uuid4, missing=uuid.uuid4)
+    target_class_name = fields.String()

    @classmethod  # This threats none values as missing
    def get_attribute(cls, attr, obj, default):