From 59f99bff394ddbab1262820fc5cc12b9d54236f7 Mon Sep 17 00:00:00 2001
From: Jan Kuipers <jan.kuipers@elastic.co>
Date: Fri, 27 Sep 2024 09:30:19 +0200
Subject: [PATCH] Fix AdaptiveAllocationsScalerTests for release

---
 .../adaptiveallocations/AdaptiveAllocationsScalerTests.java     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerTests.java b/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerTests.java
index 44aaba88c58a8..7d98aaf67a7f3 100644
--- a/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerTests.java
+++ b/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerTests.java
@@ -148,6 +148,8 @@ public void testAutoscaling_maxAllocationsSafeguard() {
     }
 
     public void testAutoscaling_scaleDownToZeroAllocations() {
+        assumeTrue("Should only run if adaptive allocations feature flag is enabled", ScaleToZeroFeatureFlag.isEnabled());
+
         AdaptiveAllocationsScaler adaptiveAllocationsScaler = new AdaptiveAllocationsScaler("test-deployment", 1);
         // 1 hour with 1 request per 1 seconds, so don't scale.
         for (int i = 0; i < 3600; i++) {