mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2024-12-15 06:09:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update view_nni.py

Browse Source 
Fix the Config Key  “NNI_JOB_DEFAULT_IMAGE” that does not exist
And fix the default priority
This commit is contained in:
jack xu 2023-05-05 15:37:58 +08:00 committed by GitHub
parent 33a26c79f3
commit 4e258e8a82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
myapp/views/view_nni.py
View File

@ -277,7 +277,7 @@ class NNI_ModelView_Base():
self.edit_form_extra_fields['job_worker_image'] = StringField(
_(self.datamodel.obj.lab('job_worker_image')),
default=json.loads(nni.job_json).get('job_worker_image',conf.get('NNI_JOB_DEFAULT_IMAGE','')) if nni and nni.job_json else conf.get('NNI_JOB_DEFAULT_IMAGE',''),
default=json.loads(nni.job_json).get('job_worker_image',conf.get('NNI_IMAGES','')) if nni and nni.job_json else conf.get('NNI_IMAGES',''),
description='工作节点镜像',
widget=BS3TextFieldWidget(),
validators=[DataRequired()]
@ -374,7 +374,7 @@ class NNI_ModelView_Base():
resource_gpu='0',
image_pull_policy=conf.get('IMAGE_PULL_POLICY','Always'),
image_pull_secrets=image_secrets,
image=conf.get('NNI_IMAGES',json.loads(nni.job_json).get('job_worker_image')) ,
image=json.loads(nni.job_json).get('job_worker_image', conf.get('NNI_IMAGES', '')),
hostAliases=conf.get('HOSTALIASES',''),
env=None,
privileged=False,