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

Add Model.project getter/setter

Browse Source
This commit is contained in:
allegroai 2022-04-27 17:32:25 +03:00
parent 938b5f0e63
commit 5318c210ab
1 changed files with 34 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
clearml/model.py
View File

@ -13,7 +13,9 @@ from pathlib2 import Path
from .utilities.config import config_dict_to_text, text_to_config_dict from .utilities.config import config_dict_to_text, text_to_config_dict
from .backend_interface.util import validate_dict, get_single_result, mutually_exclusive, exact_match_regex from .backend_interface.util import (
validate_dict, get_single_result, mutually_exclusive, exact_match_regex,
get_or_create_project, )
from .debugging.log import get_logger from .debugging.log import get_logger
from .storage.cache import CacheManager from .storage.cache import CacheManager
from .storage.helper import StorageHelper from .storage.helper import StorageHelper
@ -162,6 +164,29 @@ class BaseModel(object):
""" """
self._get_base_model().update(name=value) self._get_base_model().update(name=value)
@property
def project(self):
# type: () -> str
"""
project id of the model.
:return: project id (str).
"""
data = self._get_model_data()
return data.project
@project.setter
def project(self, value):
# type: (str) -> None
"""
Set the project ID of the model.
:param value: project ID (str).
:type value: str
"""
self._get_base_model().update(project_id=value)
@property @property
def comment(self): def comment(self):
# type: () -> str # type: () -> str
@ -621,6 +646,7 @@ class InputModel(Model):
config_dict=None, # type: Optional[dict] config_dict=None, # type: Optional[dict]
label_enumeration=None, # type: Optional[Mapping[str, int]] label_enumeration=None, # type: Optional[Mapping[str, int]]
name=None, # type: Optional[str] name=None, # type: Optional[str]
project=None, # type: Optional[str]
tags=None, # type: Optional[List[str]] tags=None, # type: Optional[List[str]]
comment=None, # type: Optional[str] comment=None, # type: Optional[str]
is_package=False, # type: bool is_package=False, # type: bool
@ -674,6 +700,7 @@ class InputModel(Model):
'person': 1 'person': 1
} }
:param str name: The name of the newly imported model. (Optional) :param str name: The name of the newly imported model. (Optional)
:param str project: The project name to add the model into. (Optional)
:param tags: The list of tags which describe the model. (Optional) :param tags: The list of tags which describe the model. (Optional)
:type tags: list(str) :type tags: list(str)
:param str comment: A comment / description for the model. (Optional) :param str comment: A comment / description for the model. (Optional)
@ -745,6 +772,12 @@ class InputModel(Model):
project_id = None project_id = None
task_id = None task_id = None
if project:
project_id = get_or_create_project(
session=task.session if task else Task._get_default_session(),
project_name=project
)
if not framework: if not framework:
# noinspection PyProtectedMember # noinspection PyProtectedMember
framework, file_ext = Framework._get_file_ext( framework, file_ext = Framework._get_file_ext(