Fix glum issue-872. (#408)

Quantco · Nov 7, 2024 · 1751b63 · 1751b63
1 parent 8a04d3f
commit 1751b63
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/tabmat/standardized_mat.py b/src/tabmat/standardized_mat.py
@@ -147,8 +147,9 @@ def sandwich(
 
         limited_shift = self.shift[cols] if cols is not None else self.shift
         limited_d = d[rows] if rows is not None else d
-        term3_and_4 = np.outer(limited_shift, d_mat + limited_shift * np.sum(limited_d))
-        res = term2 + term3_and_4
+        term3 = np.outer(limited_shift, d_mat)
+        term4 = np.outer(limited_shift, limited_shift) * np.sum(limited_d)
+        res = term2 + term3 + term4
         if isinstance(term1, sps.dia_matrix):
             idx = np.arange(res.shape[0])
             to_add = term1.data[0, :]