diff --git a/coix/oryx_test.py b/coix/oryx_test.py index 65a490b..6df092e 100644 --- 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