ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix double-escaped design when connecting OutputModel

Browse Source
This commit is contained in:
allegroai 2020-08-27 15:04:03 +03:00
parent 0d533e9013
commit b25ca9b384
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
trains/model.py
View File

@ -967,9 +967,12 @@ class OutputModel(BaseModel):
elif self._floating_data is not None:
# we copy configuration / labels if they exist, obviously someone wants them as the output base model
if _Model._unwrap_design(self._floating_data.design):
design = _Model._unwrap_design(self._floating_data.design)
if design:
if not task._get_model_config_text():
task._set_model_config(config_text=self._floating_data.design)
if not Session.check_min_api_version('2.9'):
design = self._floating_data.design
task._set_model_config(config_text=design)
else:
self._floating_data.design = _Model._wrap_design(self._task._get_model_config_text())