Fix function signatures.

I had left out the TableArgument for output.edges.