Fixup for excluding NaN in NPE

sbi-dev · Nov 3, 2022 · ecdfe09 · ecdfe09
1 parent 8afd2e4
commit ecdfe09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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