update doc examples
package io.improbable.snippet;
+import io.improbable.keanu.tensor.generic.GenericTensor;
import io.improbable.keanu.vertices.dbl.DoubleVertex;
import io.improbable.keanu.vertices.dbl.nonprobabilistic.ConstantDoubleVertex;
import io.improbable.keanu.vertices.generic.probabilistic.discrete.CategoricalVertex;
@@ -15,7 +16,7 @@ public enum MyType {
A, B, C, D
}
-public CategoricalVertex<MyType> getSelectorForMyType() {
+public CategoricalVertex<MyType, GenericTensor<MyType>> getSelectorForMyType() {
LinkedHashMap<MyType, DoubleVertex> frequency = new LinkedHashMap<>();
frequency.put(A, new ConstantDoubleVertex(0.25));
