StageGraph now properly accounts for different set of feature functions at different passes.