local test fix

Author: JuliaLinhart

tests/lc2st_test.py

@@ -197,7 +197,7 @@ def test_lc2st_false_positiv_rate(method):
 
     # use big num_train and num_epochs to obtain "good" estimator
     # (convergence of the estimator)
-    num_train = 10_000
+    num_train = 1_000
     num_epochs = 200
 
     num_cal = 1_000
@@ -215,7 +215,7 @@ def test_lc2st_false_positiv_rate(method):
     # Train the neural posterior estimators
     inference = NPE(prior, density_estimator='maf')
     inference = inference.append_simulations(theta=theta_train, x=x_train)
-    npe = inference.train(training_batch_size=1000, max_num_epochs=num_epochs)
+    npe = inference.train(training_batch_size=100, max_num_epochs=num_epochs)
 
     thetas = prior.sample((num_cal,))
     xs = simulator(thetas)

tutorials/13_diagnostics_lc2st.ipynb

@@ -277,7 +277,6 @@
     "# sample calibration data\n",
     "theta_cal = prior.sample((NUM_CAL,))\n",
     "x_cal = simulator(theta_cal)\n",
-    "# post_samples_cal = npe.sample((1,), x_cal).reshape(-1, theta_cal.shape[-1]).detach()\n",
     "post_samples_cal = posterior.sample_batched((1,), x=x_cal)[0]"
    ]
   },