clearml-server/apiserver/database/model/task/output.py
2021-01-05 16:28:49 +02:00

17 lines
458 B
Python

from mongoengine import EmbeddedDocument, StringField
from apiserver.database.fields import StrippedStringField
from apiserver.database.utils import get_options
class Result(object):
success = 'success'
failure = 'failure'
class Output(EmbeddedDocument):
destination = StrippedStringField()
model = StringField(reference_field='Model')
error = StringField(user_set_allowed=True)
result = StringField(choices=get_options(Result))