ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-01 01:26:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

OutputModel with task=None should use current task, if exists

Browse Source
This commit is contained in:
allegroai 2020-05-13 20:31:58 +03:00
parent 99f987aafb
commit 9130a06210
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
trains/model.py
View File

@ -862,7 +862,7 @@ class OutputModel(BaseModel):
def __init__(
self,
task, # type: Task
task=None, # type: Optional[Task]
config_text=None, # type: Optional[str]
config_dict=None, # type: Optional[dict]
label_enumeration=None, # type: Optional[Dict[str, int]]
@ -904,6 +904,12 @@ class OutputModel(BaseModel):
:type framework: str or Framework object
:param base_model_id: optional, model id to be reused
"""
if not task:
from .task import Task
task = Task.current_task()
if not task:
raise ValueError("task object was not provided, and no current task was found")
super(OutputModel, self).__init__(task=task)
config_text = self._resolve_config(config_text=config_text, config_dict=config_dict)