supervisor: Make LagBasedAutoScaler get current task count from the c…

…onfig
adithyachakilam · Aug 23, 2024 · cb5238e · cb5238e
1 parent 60d4317
commit cb5238e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...va/org/apache/druid/indexing/seekablestream/supervisor/autoscaler/LagBasedAutoScaler.java b/...va/org/apache/druid/indexing/seekablestream/supervisor/autoscaler/LagBasedAutoScaler.java
@@ -214,7 +214,7 @@ private int computeDesiredTaskCount(List<Long> lags)
         withinProportion, dataSource
     );
 
-    int currentActiveTaskCount = supervisor.getActiveTaskGroupsCount();
+    int currentActiveTaskCount = supervisor.getIoConfig().getTaskCount();
     int desiredActiveTaskCount;
 
     if (beyondProportion >= lagBasedAutoScalerConfig.getTriggerScaleOutFractionThreshold()) {