Update docstrings to match implementation

lampe/inference.py

@@ -221,7 +221,7 @@ class CalNRELoss(nn.Module):
     .. math::
         l & = \frac{1}{2N} \sum_{i = 1}^N
             \ell(d_\phi(\theta_i, x_i)) + \ell(1 - d_\phi(\theta_{i+1}, x_i)) \\
-          & + \lambda \max_j | \text{ECP}(1 - \alpha_j) - (1 - \alpha_j)|
+          & + \lambda 1/M \sum_{j=1}^M | \text{ECP}(1 - \alpha_j) - (1 - \alpha_j)|
 
     where :math:`\ell(p) = -\log p` is the negative log-likelihood and
     :math:`\text{ECP}(1 - \alpha_j)` is the Expected Coverage Probability at
@@ -700,7 +700,7 @@ class CalNPELoss(nn.Module):
 
     .. math::
         l & = \frac{1}{N} \sum_{i = 1}^N -\log p_\phi(\theta_i | x_i) \\
-          & + \lambda \max_j | \text{ECP}(1 - \alpha_j) - (1 - \alpha_j)|
+          & + \lambda 1/M \sum_{j=1}^M | \text{ECP}(1 - \alpha_j) - (1 - \alpha_j)|
 
     where :math:`\ell(p) = -\log p` is the negative log-likelihood and
     :math:`\text{ECP}(1 - \alpha_j)` is the Expected Coverage Probability at