Fix line too long linting error.

pyro/ops/stats.py

@@ -293,7 +293,8 @@ def weighed_quantile(input, probs, log_weights, dim=0):
     weights = weights - weights.min(dim, keepdim=True)[0]
     weights = weights / weights.max(dim, keepdim=True)[0]
     # Calculate indices
-    indices_above = (weights[..., None] <= probs).sum(dim, keepdim=True).swapaxes(dim, -1).clamp(max=input.size(dim) - 1)[..., 0]
+    indices_above = (weights[..., None] <= probs).sum(dim, keepdim=True).swapaxes(dim, -1)
+    indices_above = indices_above.clamp(max=input.size(dim) - 1)[..., 0]
     indices_below = (indices_above - 1).clamp(min=0)
     # Calculate below and above qunatiles
     quantiles_below = sorted_input.gather(dim, indices_below)