Skip to content

fix: rebaseline JAX point-source likelihood literals after noise-scale change#78

Merged
Jammy2211 merged 1 commit intomainfrom
feature/cluster-c-point-source-rebaseline
May 8, 2026
Merged

fix: rebaseline JAX point-source likelihood literals after noise-scale change#78
Jammy2211 merged 1 commit intomainfrom
feature/cluster-c-point-source-rebaseline

Conversation

@Jammy2211
Copy link
Copy Markdown
Collaborator

Summary

Rebaseline three JAX point-source likelihood literals after the realistic noise change in 931a381 (positions_noise_map 0.2" → 0.005"). The committed seed dataset was last regenerated under the old noise scale, so any fresh simulator run was producing likelihoods that did not match the hardcoded constants. This commit regenerates the seed dataset and updates the literals so both canonical-checkout runs and clean re-simulations pass.

Scripts Changed

  • scripts/jax_likelihood_functions/point_source/image_plane.py — rebaseline expected_likelihood from 1.313508 to -83.38049778
  • scripts/jax_likelihood_functions/point_source/point.py — rebaseline expected_likelihood from 1.313508 to -83.38049778 (identical to image_plane.py; both scripts load the same dataset and evaluate at the same prior medians)
  • scripts/jax_likelihood_functions/point_source/source_plane.py — rebaseline EXPECTED_VMAP_LOG_LIKELIHOOD_SOURCE_PLANE to -331481.25978149 and EXPECTED_EAGER_LOG_LIKELIHOOD_SOURCE_PLANE to -331481.26508536364. The 1664x magnitude jump matches chi-squared rescaling under a 40x position-noise reduction (1/sigma² scaling).

Also regenerates dataset/point_source/simple/{point_dataset_positions_only,tracer}.json so the on-disk seed dataset, the simulator code, and the literals are consistent. Without the regeneration, anyone who deletes the local dataset (forcing should_simulate to fire) would hit the same failure on next run.

Test Plan

  • Smoke tests pass for autolens_workspace_test
  • All three scripts exit 0 (verified locally; logs at /tmp/cluster_c_*_final.log)

🤖 Generated with Claude Code

…e change

Regenerate dataset/point_source/simple/ seed data and update the
expected_likelihood literals in scripts/jax_likelihood_functions/point_source/
{image_plane,point,source_plane}.py to match the realistic position-noise
scale (0.005") introduced in 931a381. The committed seed dataset was last
regenerated under the old 0.2" noise, so fresh simulator runs were
producing likelihoods that did not match the hardcoded constants.

- image_plane.py / point.py: 1.313508 → -83.38049778
- source_plane.py vmap:      -199.1555813 → -331481.25978149
- source_plane.py eager:     -199.1555813 → -331481.26508536364

The 1664x magnitude jump on source_plane.py is consistent with chi-squared
rescaling under a 40x position-noise reduction (1/sigma² scaling).
@Jammy2211 Jammy2211 added the pending-release PR queued for the next release build label May 8, 2026
@Jammy2211 Jammy2211 merged commit c8d4b67 into main May 8, 2026
2 of 4 checks passed
@Jammy2211 Jammy2211 deleted the feature/cluster-c-point-source-rebaseline branch May 8, 2026 08:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pending-release PR queued for the next release build

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant