Skip to content

No need to wrap inputs in jnp.array#3

Merged
TonyZhou729 merged 2 commits intomainfrom
noasarray
Feb 12, 2026
Merged

No need to wrap inputs in jnp.array#3
TonyZhou729 merged 2 commits intomainfrom
noasarray

Conversation

@cgiovanetti
Copy link
Collaborator

This wraps all params.get()'s in jnp.array internally, so the inputs to jitted functions are still wrapped in np.array without the user needing to do so in their input params dictionary. I have verified across a couple scenarios (varying LCDM parameters, switching to bbn_type='LINX') that the cached compiled code is still used when non-wrapped float inputs change value.

@TonyZhou729 TonyZhou729 merged commit dd430cd into main Feb 12, 2026
1 check passed
@cgiovanetti cgiovanetti deleted the noasarray branch February 12, 2026 17:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants