From 5e926c294ba6b859cdc25b09025b168e963c58f3 Mon Sep 17 00:00:00 2001
From: Nathalie Jonathan <nathhjo@amazon.com>
Date: Thu, 14 Nov 2024 16:23:20 -0800
Subject: [PATCH] Added missing model states

Signed-off-by: Nathalie Jonathan <nathhjo@amazon.com>
---
 spec/schemas/ml._common.yaml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/spec/schemas/ml._common.yaml b/spec/schemas/ml._common.yaml
index 39c8cb97e..2eb2536c2 100644
--- a/spec/schemas/ml._common.yaml
+++ b/spec/schemas/ml._common.yaml
@@ -114,8 +114,13 @@ components:
           type: string
           description: The model state.
           enum:
-            - DEPLOY_FAILED
             - REGISTERING
+            - REGISTERED
+            - DEPLOYING
+            - DEPLOYED
+            - PARTIALLY_DEPLOYED
+            - UNDEPLOYED
+            - DEPLOY_FAILED
         planning_worker_node_count:
           type: number
           description: The planning worker node count.