From 0f7393e0dfb4dada7582cbda4b8a3f665ec7bf13 Mon Sep 17 00:00:00 2001
From: Jan Boelts <jan.boelts@mailbox.org>
Date: Fri, 16 Feb 2024 11:30:22 +0100
Subject: [PATCH] add maf_rqs and kwargs to neural net tests.

---
 tests/sbiutils_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/sbiutils_test.py b/tests/sbiutils_test.py
index e6e08a2f8..e773ba277 100644
--- a/tests/sbiutils_test.py
+++ b/tests/sbiutils_test.py
@@ -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: