From 4080ab6f55c225fbdab950ac89c021004fa2b768 Mon Sep 17 00:00:00 2001
From: Gitea <gitea@fake.local>
Date: Mon, 18 Jul 2022 20:32:07 +0800
Subject: [PATCH] fix issue

---
 templates/custom/wait_count_train.tmpl | 4 ++--
 templates/repo/debugjob/index.tmpl     | 4 ++--
 web_src/js/features/cloudrbanin.js     | 4 ++++
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/templates/custom/wait_count_train.tmpl b/templates/custom/wait_count_train.tmpl
index ef6c4ed61..a0e130c61 100644
--- a/templates/custom/wait_count_train.tmpl
+++ b/templates/custom/wait_count_train.tmpl
@@ -3,13 +3,13 @@
 	{{$gpuQueue := 0}}
 	{{range $k,$v :=.gpu_types}}
 	{{if eq $k 0}}
-	{{ $queue := $v.Queue }}
+	{{ $queue = $v.Queue }}
 	{{ end }}
 	{{ end }}
 
 	{{ range $k,$v :=.QueuesDetail }}
 	{{if eq $k $queue}}
-	{{$gpuQueue :=$v}}
+	{{$gpuQueue =$v}}
 	{{ end }}
 	{{ end }}
 	<i
diff --git a/templates/repo/debugjob/index.tmpl b/templates/repo/debugjob/index.tmpl
index 14a74ef31..043816e9f 100755
--- a/templates/repo/debugjob/index.tmpl
+++ b/templates/repo/debugjob/index.tmpl
@@ -386,7 +386,7 @@
                                             {{$.CsrfTokenHtml}}
                                             {{if .CanDel}}
                                             <a id="ai-stop-{{.Cloudbrain.ID}}"
-                                                class='ui basic ai_stop {{if eq .Status "STOPPED" "FAILED" "START_FAILED" "STOPPING"  "CREATING"  "STARTING" "SUCCEEDED"}}disabled {{else}}blue {{end}}button'
+                                                class='ui basic ai_stop {{if eq .Status "STOPPED" "FAILED" "START_FAILED" "STOPPING"  "CREATING"  "STARTING" "SUCCEEDED" "CREATE_FAILED"}}disabled {{else}}blue {{end}}button'
                                                 data-repopath="{{$.RepoLink}}{{if eq .ComputeResource "CPU/GPU"}}/cloudbrain{{else}}/modelarts/notebook{{end}}/{{.Cloudbrain.ID}}/stop"
                                                 data-jobid="{{.Cloudbrain.ID}}">
                                                 {{$.i18n.Tr "repo.stop"}}
@@ -405,7 +405,7 @@
                                             {{$.CsrfTokenHtml}}
                                             {{if .CanDel}}
                                             <a id="ai-delete-{{.Cloudbrain.ID}}"
-                                                class='ui basic ai_delete {{if eq .Status "STOPPED" "FAILED" "START_FAILED" "SUCCEEDED"}}blue {{else}}disabled {{end}}button'
+                                                class='ui basic ai_delete {{if eq .Status "STOPPED" "FAILED" "START_FAILED" "SUCCEEDED" "CREATE_FAILED"}}blue {{else}}disabled {{end}}button'
                                                 style="border-radius: .28571429rem;">
                                                 {{$.i18n.Tr "repo.delete"}}
                                             </a>
diff --git a/web_src/js/features/cloudrbanin.js b/web_src/js/features/cloudrbanin.js
index 95d88fc0b..b7a9e4eef 100644
--- a/web_src/js/features/cloudrbanin.js
+++ b/web_src/js/features/cloudrbanin.js
@@ -123,6 +123,7 @@ export default async function initCloudrain() {
               "KILLED",
               "COMPLETED",
               "SUCCEEDED",
+              "CREATE_FAILED"
             ].includes(status)
           ) {
             $("#ai-delete-" + ID)
@@ -166,6 +167,7 @@ export default async function initCloudrain() {
           "START_FAILED",
           "SUCCEEDED",
           "STOPPED",
+          "CREATE_FAILED",
         ].includes(status)
       ) {
         return;
@@ -177,6 +179,7 @@ export default async function initCloudrain() {
         "KILLING",
         "COMPLETED",
         "SUCCEEDED",
+        "CREATE_FAILED",
         "STOPPED",
       ];
       $.get(
@@ -385,6 +388,7 @@ export default async function initCloudrain() {
           "KILLED",
           "COMPLETED",
           "SUCCEEDED",
+          "CREATE_FAILED",
         ].includes(data.JobStatus)
       ) {
         $("#ai-delete-" + ID)