fixed formatting

timnaher · timnaher · commit f4abd1b8eccc · 2026-03-02T11:59:31.000+01:00
diff --git a/spd_learn/functional/regularize.py b/spd_learn/functional/regularize.py
@@ -79,7 +79,8 @@ def ledoit_wolf(
     alpha = torch.sigmoid(shrinkage)
 
     return (1 - alpha)[..., None, None] * covariances + alpha[..., None, None] * (
-        mu[..., None] * shrink_mat.to(device=covariances.device, dtype=covariances.dtype)
+        mu[..., None]
+        * shrink_mat.to(device=covariances.device, dtype=covariances.dtype)
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,8 @@ def ledoit_wolf(`
`79`	`79`	`alpha = torch.sigmoid(shrinkage)`
`80`	`80`
`81`	`81`	`return (1 - alpha)[..., None, None] * covariances + alpha[..., None, None] * (`
`82`		`- mu[..., None] * shrink_mat.to(device=covariances.device, dtype=covariances.dtype)`
	`82`	`+ mu[..., None]`
	`83`	`+ * shrink_mat.to(device=covariances.device, dtype=covariances.dtype)`
`83`	`84`	`)`
`84`	`85`
`85`	`86`