sbi/inference/snle/snle_a.py