Skip to content

Fix EnsembleSampler second split to use complementary inactive chains#2189

Open
amifalk wants to merge 1 commit into
pyro-ppl:masterfrom
amifalk:ensemble-sampler-bf
Open

Fix EnsembleSampler second split to use complementary inactive chains#2189
amifalk wants to merge 1 commit into
pyro-ppl:masterfrom
amifalk:ensemble-sampler-bf

Conversation

@amifalk
Copy link
Copy Markdown
Contributor

@amifalk amifalk commented May 16, 2026

The second sub-iteration in EnsembleSampler.sample incorrectly updated the second half of chains against itself instead of the first half. Add a deterministic regression test that fails under the previous split logic and verifies complementary-half updates.

Original Author: @jackkohm

@amifalk
Copy link
Copy Markdown
Contributor Author

amifalk commented May 16, 2026

The original PR (#2135) seems dead but was good to go save for failing lint.

@amifalk
Copy link
Copy Markdown
Contributor Author

amifalk commented May 16, 2026

@Qazalbash looks like this is failing because of the new toml formatting checks?

@Qazalbash
Copy link
Copy Markdown
Collaborator

Thanks @amifalk. I have proposed a fix. You can merge the master into your branch after #2190 is merged.

The second sub-iteration in EnsembleSampler.sample incorrectly updated the
second half of chains against itself instead of the first half. Add a
deterministic regression test that fails under the previous split logic and
verifies complementary-half updates.
@amifalk amifalk force-pushed the ensemble-sampler-bf branch from 5a81279 to 30e087e Compare May 17, 2026 12:16
@Qazalbash
Copy link
Copy Markdown
Collaborator

This looks identical to #2135.

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.

3 participants