Wait for trained model to be fully allocated (#145)

pquentin · web-flow · commit 16c39509a6d8 · 2024-10-14T19:12:18.000+04:00
diff --git a/tests/machine_learning/20_trained_model_serverless.yml b/tests/machine_learning/20_trained_model_serverless.yml
@@ -52,7 +52,8 @@ teardown:
       ml.start_trained_model_deployment:
         model_id: test_model
         priority: "low"
-        wait_for: started
+        wait_for: fully_allocated
+        timeout: "60s"
   - match: { assignment.task_parameters.model_id: test_model }
   - do:
       ml.update_trained_model_deployment:
