ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-06-26 23:15:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Task.container support

Browse Source
This commit is contained in:
allegroai
2021-05-03 17:48:01 +03:00
parent 5a438e8435
commit f4d5168a20
5 changed files with 146 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
apiserver/database/model/task/task.py
View File

@@ -125,7 +125,6 @@ class Execution(EmbeddedDocument, ProperDictMixin):
model_labels = ModelLabels()
framework = StringField()
artifacts: Dict[str, Artifact] = SafeMapField(field=EmbeddedDocumentField(Artifact))
docker_cmd = StringField()
queue = StringField(reference_field="Queue")
""" Queue ID where task was queued """
@@ -250,7 +249,7 @@ class Task(AttributedDocument):
configuration = SafeMapField(field=EmbeddedDocumentField(ConfigurationItem))
runtime = SafeDictField(default=dict)
models: Models = EmbeddedDocumentField(Models, default=Models)
docker_init_script = StringField()
container = SafeMapField(field=StringField(default=""))
def get_index_company(self) -> str:
"""