fix oryx test

jax-ml · Oct 24, 2024 · 8685b10 · 8685b10
1 parent a905856
commit 8685b10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/coix/oryx_test.py b/coix/oryx_test.py
@@ -113,7 +113,7 @@ def expected_fn(x):
 
 def test_observed():
   def model(a):
-    return coryx.rv(dist.Delta(2.0, 3.0), obs=1.0, name="x") + a
+    return coryx.rv(dist.Delta(a, 3.0), obs=1.0, name="x") + a
 
   _, trace, _ = coix.traced_evaluate(model)(2.0)
   assert "x" in trace