[ML] Fix random threads_per_allocation in unit test (#89588)

Closes #89584
elastic · Aug 25, 2022 · 16e4cb1 · 16e4cb1
1 parent 8cac490
commit 16e4cb1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.../java/org/elasticsearch/xpack/core/ml/action/StartTrainedModelDeploymentRequestTests.java b/.../java/org/elasticsearch/xpack/core/ml/action/StartTrainedModelDeploymentRequestTests.java
@@ -53,7 +53,7 @@ public static Request createRandom() {
             request.setWaitForState(randomFrom(AllocationStatus.State.values()));
         }
         if (randomBoolean()) {
-            request.setThreadsPerAllocation(randomIntBetween(1, 8));
+            request.setThreadsPerAllocation(randomFrom(1, 2, 4, 8, 16, 32));
         }
         if (randomBoolean()) {
             request.setNumberOfAllocations(randomIntBetween(1, 8));
@@ -150,7 +150,6 @@ public void testValidate_GivenQueueCapacityIsNegative() {
         assertThat(e.getMessage(), containsString("[queue_capacity] must be a positive integer"));
     }
 
-    @AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/89584")
     public void testValidate_GivenQueueCapacityIsAtLimit() {
         Request request = createRandom();
         request.setQueueCapacity(1_000_000);