fix docstring of WeightedGradient