Skip to content

Commit

Permalink
add maf_rqs and kwargs to neural net tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
janfb committed Feb 16, 2024
1 parent b96a8d9 commit 0f7393e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/sbiutils_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -415,13 +415,15 @@ def test_z_scoring_structured(z_x, z_theta, builder):

# API tests
if builder in [likelihood_nn, posterior_nn]:
for model in ["mdn", "made", "maf", "nsf"]:
for model in ["mdn", "made", "maf", "nsf", "maf_rqs"]:
net = builder(
model,
z_score_theta=z_theta,
z_score_x=z_x,
hidden_features=2,
num_transforms=1,
# test passing kwargs.
**{"num_bins": 10, "tail_bound": 1.0, "dtype": torch.float16},
)
assert net(t_batch, t_batch)
else:
Expand Down

0 comments on commit 0f7393e

Please sign in to comment.