log model_state

Signed-off-by: kalyan <[email protected]>
rawwar · Oct 12, 2023 · 68b237a · 68b237a
1 parent 0338bd8
commit 68b237a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opensearch_py_ml/ml_commons/ml_commons_client.py b/opensearch_py_ml/ml_commons/ml_commons_client.py
@@ -316,7 +316,7 @@ def load_model(self, model_id: str, wait_until_loaded: bool = True) -> object:
                 print("Model deployed only partially")
             # TODO: need to add the test case later for this line
             else:
-                raise Exception("Model deployment failed")
+                raise Exception("Model deployment failed. Model state: " + str(model_state))
 
         return self._get_task_info(task_id)