From 5a41ec79b841f956a2896caad34e2c1bb9a972ed Mon Sep 17 00:00:00 2001 From: Quentin Pradet Date: Mon, 14 Oct 2024 16:38:49 +0400 Subject: [PATCH] Wait for trained model to be fully allocated --- tests/machine_learning/20_trained_model_serverless.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/machine_learning/20_trained_model_serverless.yml b/tests/machine_learning/20_trained_model_serverless.yml index 4f22fe8..27f48a3 100644 --- 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: