Skip to content

Commit

Permalink
Tests for measurable branches and deny discrete
Browse files Browse the repository at this point in the history
  • Loading branch information
shreyas3156 committed Aug 8, 2023
1 parent 9053937 commit d261140
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/logprob/test_censoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,7 +287,7 @@ def test_switch_encoding_both_branches():
],
)
def test_switch_encoding_one_branch_measurable(measurable_idx, test_values, exp_logp):
x_rv = pt.random.normal(0.5, 1) # should not be defined again ideally
x_rv = pt.random.normal(0.5, 1)
branches = (1, x_rv) if measurable_idx == 1 else (x_rv, 1)

y_rv = pt.switch(x_rv < 1, *branches)
Expand Down

0 comments on commit d261140

Please sign in to comment.