Small changes for amp subgraphs #11138

Job	Run time
Build in release mode	5m 16s
Run integration tests	4m 49s
Subgraph Runner integration tests	3m 46s
Run unit tests	3m 23s
Check rustfmt style	27s
Clippy linting	5m 24s
	23m 5s

Provide feedback