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

Fix Task.create() standalone mode

Browse Source
This commit is contained in:
allegroai 2020-12-30 16:40:47 +02:00
parent 486520abfd
commit 9f0d7b5eb8
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clearml/backend_interface/task/populate.py
View File

@ -189,6 +189,10 @@ class CreateAndPopulate(object):
task_state['script']['diff'] = '' task_state['script']['diff'] = ''
task_state['script']['working_dir'] = cwd or '.' task_state['script']['working_dir'] = cwd or '.'
task_state['script']['entry_point'] = entry_point task_state['script']['entry_point'] = entry_point
else:
# standalone task
task_state['script']['entry_point'] = self.script
task_state['script']['working_dir'] = '.'
# update requirements # update requirements
reqs = [] reqs = []
@ -219,7 +223,7 @@ class CreateAndPopulate(object):
"Use --requirements or --packages".format(reqs_txt_file.as_posix())) "Use --requirements or --packages".format(reqs_txt_file.as_posix()))
if self.add_task_init_call: if self.add_task_init_call:
script_entry = os.path.abspath('/' + task_state['script']['working_dir'] + script_entry = os.path.abspath('/' + task_state['script'].get('working_dir', '.') +
'/' + task_state['script']['entry_point']) '/' + task_state['script']['entry_point'])
idx_a = 0 idx_a = 0
# find the right entry for the patch if we have a local file (basically after __future__ # find the right entry for the patch if we have a local file (basically after __future__
@ -264,7 +268,7 @@ class CreateAndPopulate(object):
"from clearml import Task\n" \ "from clearml import Task\n" \
"Task.init()\n\n" "Task.init()\n\n"
task_state['script']['diff'] = task_init_patch + task_state['script']['diff'] task_state['script']['diff'] = task_init_patch + task_state['script'].get('diff', '')
# set base docker image if provided # set base docker image if provided
if self.docker: if self.docker: