Skip to content

Log power spectrum diagnostics in inference entrypoint#1078

Merged
spencerkclark merged 2 commits into
mainfrom
feature/power-spectrum-diagnostics-inference
Apr 23, 2026
Merged

Log power spectrum diagnostics in inference entrypoint#1078
spencerkclark merged 2 commits into
mainfrom
feature/power-spectrum-diagnostics-inference

Conversation

@spencerkclark
Copy link
Copy Markdown
Member

This PR implements what is necessary to add the SphericalPowerSpectrumAggregator to the InferenceAggregator, so that we can now log power spectra in runs launched with the inference entrypoint.

Changes:

  • Adds get_logs and get_dataset methods to the SphericalPowerSpectrumAggregator.

  • Adds SphericalPowerSpectrumAggregator to the InferenceAggregator.

  • Tests added

Resolves #1077

Extends `SphericalPowerSpectrumAggregator` with `get_logs`/`get_dataset`
so it satisfies the `_Aggregator` protocol, then wires it into
`InferenceAggregator` as a summary aggregator mirroring the evaluator.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
@spencerkclark spencerkclark force-pushed the feature/power-spectrum-diagnostics-inference branch from 2d698c8 to 2e975bb Compare April 23, 2026 15:31
Copy link
Copy Markdown
Member

@jpdunc23 jpdunc23 left a comment

Choose a reason for hiding this comment

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

LGTM!

@spencerkclark spencerkclark merged commit 66e5a0e into main Apr 23, 2026
7 checks passed
@spencerkclark spencerkclark deleted the feature/power-spectrum-diagnostics-inference branch April 23, 2026 19:05
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.

Power spectrum diagnostics are not logged for rollouts with the inference entrypoint

2 participants