I’m going to close this topic out, as it is a few years old and there is a working solution in the Python embedded above as noted by @MartinSpence.
@Rodolfo_Jordan if you still have issues post a small sample Revit file and your working graph to a new topic, linking back here, and the community will do what we can to help you out.