From ecdfe091762490e439ce4892ed9086fa7e2c7063 Mon Sep 17 00:00:00 2001
From: michaeldeistler <michael.deistler@uni-tuebingen.de>
Date: Thu, 3 Nov 2022 17:44:01 +0100
Subject: [PATCH] Fixup for excluding NaN in NPE

---
 sbi/inference/snpe/snpe_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sbi/inference/snpe/snpe_base.py b/sbi/inference/snpe/snpe_base.py
index 5428d7963..806bb5e86 100644
--- a/sbi/inference/snpe/snpe_base.py
+++ b/sbi/inference/snpe/snpe_base.py
@@ -138,7 +138,7 @@ def append_simulations(
                 current_round = max(self._data_round_index) + 1
 
         if exclude_invalid_x is None:
-            if current_round > 0:
+            if current_round == 0:
                 exclude_invalid_x = True
             else:
                 exclude_invalid_x = False