com.yahoo.tensor.evaluation.MapEvaluationContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vespajlib Show documentation

Library for use in Java components of Vespa. Shared code which do not fit anywhere else.

There is a newer version: 8.441.21

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.tensor.evaluation;

import com.yahoo.tensor.Tensor;
import com.yahoo.tensor.TensorType;

import java.util.HashMap;

/**
 * @author bratseth
 */
public class MapEvaluationContext implements EvaluationContext {

    private final java.util.Map bindings = new HashMap<>();

    public void put(String name, Tensor tensor) { bindings.put(name, tensor); }

    @Override
    public TensorType getType(String name) {
        Tensor tensor = bindings.get(name);
        if (tensor == null) return null;
        return tensor.type();
    }

    @Override
    public TensorType getType(NAMETYPE name) { return getType(name.name()); }

    @Override
    public Tensor getTensor(String name) { return bindings.get(name); }

    @Override
    public String resolveBinding(String name) { return name; }

}