ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix auto-detect git repository

Browse Source
This commit is contained in:
allegroai 2019-08-01 12:54:11 +03:00
parent bcaf4ce6c8
commit fb7dfc49a9
2 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains/backend_interface/task/task.py
View File

@ -211,10 +211,11 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
# overwrite it before we have a chance to call edit) # overwrite it before we have a chance to call edit)
self._edit(script=result.script) self._edit(script=result.script)
self.reload() self.reload()
self._update_requirements(result.script.get('requirements') if result.script.get('requirements') else '') self._update_requirements(result.script.get('requirements') if result.script and
result.script.get('requirements') else '')
check_package_update_thread.join() check_package_update_thread.join()
except Exception as e: except Exception as e:
get_logger('task').warning(str(e)) get_logger('task').debug(str(e))
def _auto_generate(self, project_name=None, task_name=None, task_type=TaskTypes.training): def _auto_generate(self, project_name=None, task_name=None, task_type=TaskTypes.training):
created_msg = make_message('Auto-generated at %(time)s by %(user)s@%(host)s') created_msg = make_message('Auto-generated at %(time)s by %(user)s@%(host)s')

24
trains/task.py
View File

@ -334,17 +334,19 @@ class Task(_Task):
if not default_project_name or not default_task_name: if not default_project_name or not default_task_name:
# get project name and task name from repository name and entry_point # get project name and task name from repository name and entry_point
result = ScriptInfo.get(create_requirements=False, check_uncommitted=False) result = ScriptInfo.get(create_requirements=False, check_uncommitted=False)
if result: if not default_project_name:
if not default_project_name: # noinspection PyBroadException
try: try:
default_project_name = re.sub(r"\.git$", "", result.script.get('repository')) or "Untitled" parts = result.script['repository'].split('/')
except TypeError: default_project_name = (parts[-1] or parts[-2]).replace('.git', '') or 'Untitled'
default_project_name = 'Untitled' except Exception:
if not default_task_name: default_project_name = 'Untitled'
try: if not default_task_name:
default_task_name = Path(result.script.get("entry_point")).stem # noinspection PyBroadException
except TypeError: try:
pass default_task_name = os.path.splitext(os.path.basename(result.script['entry_point']))[0]
except Exception:
pass
# if we force no task reuse from os environment # if we force no task reuse from os environment
if DEV_TASK_NO_REUSE.get() or not reuse_last_task_id: if DEV_TASK_NO_REUSE.get() or not reuse_last_task_id: