ml

henningandersen · May 24, 2024 · 6554796 · 6554796
1 parent 8de5f11
commit 6554796
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...va/org/elasticsearch/xpack/ml/inference/assignment/TrainedModelAssignmentNodeService.java b/...va/org/elasticsearch/xpack/ml/inference/assignment/TrainedModelAssignmentNodeService.java
@@ -13,6 +13,7 @@
 import org.elasticsearch.action.ActionListener;
 import org.elasticsearch.action.search.SearchPhaseExecutionException;
 import org.elasticsearch.action.support.PlainActionFuture;
+import org.elasticsearch.action.support.UnsafePlainActionFuture;
 import org.elasticsearch.action.support.master.AcknowledgedResponse;
 import org.elasticsearch.cluster.ClusterChangedEvent;
 import org.elasticsearch.cluster.ClusterState;
@@ -205,7 +206,9 @@ void loadQueuedModels() {
             if (stopped) {
                 return;
             }
-            final PlainActionFuture<TrainedModelDeploymentTask> listener = new PlainActionFuture<>();
+            final PlainActionFuture<TrainedModelDeploymentTask> listener = new UnsafePlainActionFuture<>(
+                MachineLearning.UTILITY_THREAD_POOL_NAME
+            );
             try {
                 deploymentManager.startDeployment(loadingTask, listener);
                 // This needs to be synchronous here in the utility thread to keep queueing order