Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feature/jax #64

Merged
merged 10 commits into from
Jan 29, 2024
Merged

feature/jax #64

merged 10 commits into from
Jan 29, 2024

Conversation

rhayes777
Copy link
Collaborator

Changes to support new jax compliant autoarray

y = data.apply_mask(mask=np.invert(fpr_mask)).binned_across_columns
y = data.apply_mask(mask=fpr_mask).binned_across_columns
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't really know what happened here but for some reason the test only passed without the inversion. If it looks dodgy I'll investigate further

Comment on lines +141 to +144
mask=aa.Mask1D(
sum(mask_1d_list) == len(mask_1d_list),
pixel_scales=mask_1d_list[0].pixel_scales,
),
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Masks aren't arrays anymore so this had to be cast back explicitly. I might try to make it implicit again

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actually I'm not totally sure what the point of this operation is

@rhayes777 rhayes777 merged commit bd92881 into main Jan 29, 2024
1 of 9 checks passed
@rhayes777 rhayes777 deleted the feature/jax branch January 29, 2024 08:48
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