def _build_bayes_net(
builder: JavaObject, net: Union[BayesNet, Vertex]
) -> Tuple[JavaObject, Union[BayesNet, Vertex]]:
if not (isinstance(net, BayesNet) or isinstance(net, Vertex)):
raise TypeError(
"net must be a Vertex or a BayesNet. Was given {}".format(
type(net)))
elif isinstance(net, Vertex):
net = BayesNet(net.get_connected_graph())
return builder.bayesianNetwork(net.unwrap()), net
