There should be a space after the colon. Also I think unless there is a long list of arguments it's fine to just do a normal wrap. I'm not sure exactly what the right way to wrap this is but one idea is proposed below:

```
def extractLongDistribution(stage:StageCompleted, getMetric: (TaskInfo,TaskMetrics) => Option[Long]): 
  Option[Distribution] = {
```