diff options
Diffstat (limited to 'model-evaluation')
| -rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java index 1bcd6363d2d..18d81e45819 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java +++ b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java @@ -4,6 +4,8 @@ package ai.vespa.models.handler; import ai.vespa.models.evaluation.FunctionEvaluator; import ai.vespa.models.evaluation.Model; import ai.vespa.models.evaluation.ModelsEvaluator; +import com.yahoo.component.annotation.Inject; +import com.yahoo.component.provider.ComponentRegistry; import com.yahoo.container.jdisc.HttpRequest; import com.yahoo.container.jdisc.HttpResponse; import com.yahoo.container.jdisc.ThreadedHttpRequestHandler; @@ -39,6 +41,14 @@ public class ModelsEvaluationHandler extends ThreadedHttpRequestHandler { private final ModelsEvaluator modelsEvaluator; + @Inject + public ModelsEvaluationHandler(ComponentRegistry<ModelsEvaluator> registry, + Executor executor) + { + this(registry.getComponent("default/component/" + ModelsEvaluator.class.getName()), + executor); + } + public ModelsEvaluationHandler(ModelsEvaluator modelsEvaluator, Executor executor) { super(executor); this.modelsEvaluator = modelsEvaluator; |