ai.vespa.rankingexpression.importer.configmodelview.ImportedMlFunction Maven / Gradle / Ivy

Go to download
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.rankingexpression.importer.configmodelview;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;

/**
 * An imported function of an imported machine-learned model
 *
 * @author bratseth
 */
public class ImportedMlFunction {

    private final String name;
    private final List arguments;
    private final Map argumentTypes;
    private final String  expression;
    private final Optional returnType;

    public ImportedMlFunction(String name, List arguments, String expression,
                              Map argumentTypes, Optional returnType) {
        this.name = name;
        this.arguments = Collections.unmodifiableList(arguments);
        this.expression = expression;
        this.argumentTypes = Collections.unmodifiableMap(argumentTypes);
        this.returnType = returnType;
    }

    public String name() { return name; }
    public List arguments() { return arguments; }
    public Map argumentTypes() { return argumentTypes; }
    public String expression() { return expression; }
    public Optional returnType() { return returnType; }

}