this commit works with the corresponding commit on https://gitlab.laas.fr/roc/tim-luchterhand/nn-heuristic-implementation. The GNN works on the full topology extracted from the transitive closure of the scheduler. It predicts values only for edges in the graph.