Why not just pass in the preprocessed graph directly?