Annotate the Constructor for the ConstantInt
@@ -5,6 +5,7 @@ import io.improbable.keanu.annotation.ExportVertexToPythonBindings;
import io.improbable.keanu.network.NetworkWriter;
import io.improbable.keanu.tensor.intgr.IntegerTensor;
import io.improbable.keanu.vertices.ConstantVertex;
+import io.improbable.keanu.vertices.LoadVertexValue;
import io.improbable.keanu.vertices.NonProbabilistic;
import io.improbable.keanu.vertices.SaveableVertex;
import io.improbable.keanu.vertices.dbl.KeanuRandom;
@@ -13,7 +14,7 @@ import io.improbable.keanu.vertices.intgr.IntegerVertex;
public class ConstantIntegerVertex extends IntegerVertex implements ConstantVertex, SaveableVertex, NonProbabilistic<IntegerTensor> {
@ExportVertexToPythonBindings
- public ConstantIntegerVertex(IntegerTensor constant) {
+ public ConstantIntegerVertex(@LoadVertexValue IntegerTensor constant) {
super(constant.getShape());
setValue(constant);
}
