Fix inv

normal-computing · May 29, 2024 · 6b9ed8f · 6b9ed8f
1 parent 35bcfa4
commit 6b9ed8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thermox/linalg.py b/thermox/linalg.py
@@ -69,7 +69,7 @@ def inv(
     ts = jnp.arange(burnin, burnin + num_samples) * dt
     b = jnp.zeros(A.shape[0])
     x0 = jnp.zeros_like(b)
-    samples = sample_identity_diffusion(key, ts, x0, A, b, 2 * jnp.eye(A.shape[0]))
+    samples = sample(key, ts, x0, A, b, 2 * jnp.eye(A.shape[0]))
     return jnp.cov(samples.T)