Update PolicyGenerator to work with updated Ocaml-topology interface