Assertion error on `assert jnp.isfinite(x).all()`

[sagitaninta]

I was trying to fit phlash in different regions of the chromosome of my own data set. As my vcf already filtered for all the problematic sites, the region parameter does not matter, but it asks me for a region so I specify the entire chromosome (e.g., chr23_c = phlash.contig(chr23_path, samples=["DlunBCh_2230"], region="23:1-48866549") to run the entire chromosome 23).

However it barely runs when I specify the entire chromosome as shown below

Still the same when I specify half the size of it

Run quite well when I specify only 1 Mbp in the region, but then end abruptly at assert jnp.isfinite(x).all()

And similarly at 5 Mbp but faster (in ending haha)

Would you know what cause this behavior? Does it not like missing data? How should we better feed the region if the VCF is already filtered for all the SNPs that should be masked?

Many thanks in advance for your time and kind assistance!