Here's where I put the metrics from the shuffle fetches directly into the Task -- I can only do this b/c I made TaskContext keep a reference to the Task.