Fix style

normal-computing · Jul 11, 2024 · c114633 · c114633
1 parent 919f5fb
commit c114633
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/thermox/linalg.py b/thermox/linalg.py
@@ -38,7 +38,9 @@ def solve(
         key = random.PRNGKey(0)
     ts = jnp.arange(burnin, burnin + num_samples) * dt
     x0 = jnp.zeros_like(b)
-    samples = sample_identity_diffusion(key, ts, x0, A, jnp.linalg.solve(A, b), associative_scan)
+    samples = sample_identity_diffusion(
+        key, ts, x0, A, jnp.linalg.solve(A, b), associative_scan
+    )
     return jnp.mean(samples, axis=0)