preserve binary compatibility

1 files changed, 7 insertions, 3 deletions

diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/evaluation/ExpressionOptimizer.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/evaluation/ExpressionOptimizer.java
index b82173eabd5..7060cfc2132 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/evaluation/ExpressionOptimizer.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/evaluation/ExpressionOptimizer.java
@@ -44,12 +44,16 @@ public class ExpressionOptimizer {
         return null;
     }
 
-    public OptimizationReport optimize(RankingExpression expression, ContextIndex arrayContext) {
+    public OptimizationReport optimize(RankingExpression expression, ContextIndex contextIndex) {
         OptimizationReport report = new OptimizationReport();
         // Note: Order of optimizations matter
-        gbdtOptimizer.optimize(expression, arrayContext, report);
-        gbdtForestOptimizer.optimize(expression, arrayContext, report);
+        gbdtOptimizer.optimize(expression, contextIndex, report);
+        gbdtForestOptimizer.optimize(expression, contextIndex, report);
         return report;
     }
 
+    public OptimizationReport optimize(RankingExpression expression, AbstractArrayContext arrayContext) {
+        return optimize(expression, (ContextIndex)arrayContext);
+    }
+
 }