Increase sample number for flows test.

astro-informatics · Oct 24, 2023 · 0e79082 · 0e79082
1 parent 2d97ed6
commit 0e79082
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_flow_model.py b/tests/test_flow_model.py
@@ -154,7 +154,7 @@ def test_RealNVP_gaussian():
     RealNVP = model_nf.RealNVPModel(ndim, standardize=True)
     RealNVP.fit(samples, epochs=epochs, verbose=True)
 
-    nsamples = 10000
+    nsamples = 20000
     RealNVP.temperature = 1.0
     flow_samples = RealNVP.sample(nsamples)
     sample_var = jnp.var(flow_samples, axis=0)