Change snapshot api to `graph_snapshot`.

For consistency with the python lib.