fix test

janfb · janfb · commit b058b1cffe51 · 2024-12-18T10:10:19.000Z
diff --git a/tests/mnle_test.py b/tests/mnle_test.py
@@ -405,7 +405,7 @@ def test_log_likelihood_over_iid_conditions(
         theta_and_condition = torch.cat(
             (theta, condition_o[i].repeat(num_thetas, 1)), dim=1
         )
-        x_i = x_o[:, i].reshape(num_xs, 1, -1).repeat(1, num_thetas, 1)
+        x_i = x_o[i].reshape(num_xs, 1, -1).repeat(1, num_thetas, 1)
         ll_single.append(estimator.log_prob(input=x_i, condition=theta_and_condition))
     ll_single = torch.stack(ll_single).sum(0)  # sum over trials
 

Original file line number	Diff line number	Diff line change
`@@ -405,7 +405,7 @@ def test_log_likelihood_over_iid_conditions(`
`405`	`405`	`theta_and_condition = torch.cat(`
`406`	`406`	`(theta, condition_o[i].repeat(num_thetas, 1)), dim=1`
`407`	`407`	`)`
`408`		`- x_i = x_o[:, i].reshape(num_xs, 1, -1).repeat(1, num_thetas, 1)`
	`408`	`+ x_i = x_o[i].reshape(num_xs, 1, -1).repeat(1, num_thetas, 1)`
`409`	`409`	`ll_single.append(estimator.log_prob(input=x_i, condition=theta_and_condition))`
`410`	`410`	`ll_single = torch.stack(ll_single).sum(0) # sum over trials`
`411`	`411`