fix cancel error

star-whale · May 17, 2023 · dcba68b · dcba68b
1 parent 69d0445
commit dcba68b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/controller/src/main/java/ai/starwhale/mlops/domain/task/status/TaskStatusMachine.java b/server/controller/src/main/java/ai/starwhale/mlops/domain/task/status/TaskStatusMachine.java
@@ -43,7 +43,7 @@ public class TaskStatusMachine {
             new SimpleEntry<>(PAUSED, Set.of(PREPARING, ASSIGNING, RUNNING, READY, CANCELED, SUCCESS)),
             new SimpleEntry<>(ASSIGNING, Set.of(CREATED, PREPARING, RUNNING, SUCCESS, FAIL, TO_CANCEL)),
             new SimpleEntry<>(PREPARING, Set.of(RUNNING, SUCCESS, FAIL, TO_CANCEL)),
-            new SimpleEntry<>(RUNNING, Set.of(SUCCESS, FAIL, TO_CANCEL)),
+            new SimpleEntry<>(RUNNING, Set.of(SUCCESS, FAIL, TO_CANCEL, CANCELED)),
             new SimpleEntry<>(TO_CANCEL, Set.of(CANCELLING, CANCELED, SUCCESS, FAIL)),
             new SimpleEntry<>(CANCELLING, Set.of(CANCELED, FAIL)),
             new SimpleEntry<>(CANCELED, Set.of()),