hotfixing jax bug in modes.py

eelregit / pmwd

Differentiable Cosmological Forward Model

BSD 3-Clause "New" or "Revised" License

69 stars 16 forks source link

Open maho3 opened 4 months ago

maho3 commented 4 months ago

From #27 , the main branch of pmwd doesn't work due to a bug in linear_modes.

This just uses a hack to autodetermine the isrealobj from the modes, without need to manually specify.

Also, my autoformatter fixed some PEP things, but I can remove those changes if desired.