Move graphs to a single hit_graph partial

* Make sure we've got one code path for everything
* Remove the old "special case" for category