Skip to content

perf: skip radial bins computation when PYAUTO_WORKSPACE_SMALL_DATASETS=1#258

Merged
Jammy2211 merged 1 commit intomainfrom
feature/skip-radial-bins-small-datasets
Apr 7, 2026
Merged

perf: skip radial bins computation when PYAUTO_WORKSPACE_SMALL_DATASETS=1#258
Jammy2211 merged 1 commit intomainfrom
feature/skip-radial-bins-small-datasets

Conversation

@Jammy2211
Copy link
Copy Markdown
Collaborator

Summary

Early-return a uniform size-2 array from over_sample_size_via_radial_bins_from() when PYAUTO_WORKSPACE_SMALL_DATASETS=1, avoiding the numba JIT and radial distance computation. Saves ~0.1s per call (the remaining ~0.3s of apparent overhead is actually JAX lazy import, not oversampling).

API Changes

None — internal changes only, gated behind opt-in env var.

Test Plan

  • All 722 PyAutoArray tests pass

🤖 Generated with Claude Code

…TS=1

Early-return a uniform size-2 array from
over_sample_size_via_radial_bins_from(), avoiding the numba JIT
compilation and radial distance computation on small test grids.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
@Jammy2211 Jammy2211 merged commit ff0f45e into main Apr 7, 2026
2 checks passed
@Jammy2211 Jammy2211 deleted the feature/skip-radial-bins-small-datasets branch April 7, 2026 12:53
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.

1 participant