ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-22 07:15:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix task.set_script

Browse Source
This commit is contained in:
allegroai 2021-10-21 11:58:56 +03:00
parent 5bc689c9fe
commit b443e781cb
1 changed files with 28 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
clearml/task.py
View File

@ -1878,11 +1878,12 @@ class Task(_Task):
def set_script( def set_script(
self, self,
repository=NotSet, # type: Optional[str] repository=None, # type: Optional[str]
branch=NotSet, # type: Optional[str] branch=None, # type: Optional[str]
working_dir=NotSet, # type: Optional[str] commit=None, # type: Optional[str]
entry_point=NotSet, # type: Optional[str] diff=None, # type: Optional[str]
**kwargs # type: Any working_dir=None, # type: Optional[str]
entry_point=None, # type: Optional[str]
): ):
# type: (...) -> None # type: (...) -> None
""" """
@ -1890,21 +1891,34 @@ class Task(_Task):
Examples: Examples:
task.set_script(repository='https://github.com/allegroai/clearml.git, task.set_script(repository='https://github.com/allegroai/clearml.git,
branch='master', branch='main',
working_dir='examples/reporting', working_dir='examples/reporting',
entry_point='artifacts.py') entry_point='artifacts.py')
:param repository: URL of remote repository. :param repository: Optional, URL of remote repository. use empty string ("") to clear repository entry.
:param branch: Select specific repository branch / tag. :param branch: Optional, Select specific repository branch / tag. use empty string ("") to clear branch entry.
:param working_dir: Working directory to launch the script from. :param commit: Optional, set specific git commit id. use empty string ("") to clear commit id entry.
:param entry_point: Path to execute within the repository. :param diff: Optional, set "git diff" section. use empty string ("") to clear git-diff entry.
:param working_dir: Optional, Working directory to launch the script from.
:param entry_point: Optional, Path to execute within the repository.
""" """
self.reload()
script = self.data.script script = self.data.script
script.repository = script.repository if repository is self.NotSet else repository if repository is not None:
script.branch = script.branch if branch is self.NotSet else branch script.repository = str(repository) or None
script.working_dir = script.working_dir if working_dir is self.NotSet else working_dir if branch is not None:
script.entry_point = script.entry_point if entry_point is self.NotSet else entry_point script.branch = str(branch) or None
if script.tag:
script.tag = None
if commit is not None:
script.version_num = str(commit) or None
if diff is not None:
script.diff = str(diff) or None
if working_dir is not None:
script.working_dir = str(working_dir)
if entry_point is not None:
script.entry_point = str(entry_point)
# noinspection PyProtectedMember # noinspection PyProtectedMember
self._update_script(script=script) self._update_script(script=script)