Skip to content

[codex] fix initial measurement sigma#32

Draft
vvv214 wants to merge 1 commit into
google:mainfrom
vvv214:codex/fix-initial-measurement-sigma
Draft

[codex] fix initial measurement sigma#32
vvv214 wants to merge 1 commit into
google:mainfrom
vvv214:codex/fix-initial-measurement-sigma

Conversation

@vvv214

@vvv214 vvv214 commented Jun 17, 2026

Copy link
Copy Markdown

Summary

This draft PR fixes the local AIM/MST direct-call path for initial one-way measurements.

  • convert the one-way zCDP budget share with accounting.zcdp_gaussian_sigma(...)
  • pass the resulting Gaussian sigma into common.measure_marginals_with_noise
  • add tests that intercept the helper call and verify the sigma value

Why

common.measure_marginals_with_noise interprets gdp_sigma as a Gaussian standard deviation. The direct paths in AIMMechanism and MSTMechanism were passing a zCDP budget share directly when initial_measurements was not supplied.

Validation

  • python3 -m py_compile dpsynth/discrete_mechanisms/aim.py dpsynth/discrete_mechanisms/mst.py tests/discrete_mechanisms/aim_test.py tests/discrete_mechanisms/mst_test.py
  • rg -n "gdp_sigma=.*zcdp_rho \\*|gdp_sigma=self\\.zcdp_rho \\*|gdp_sigma=zcdp_rho \\*" dpsynth/discrete_mechanisms tests/discrete_mechanisms returns no matches

Full pytest was not run locally because this checkout's test dependencies (pipeline_dp, mbi, dp_accounting, Beam/JAX stack) are not installed in the current environment, and the local Python is 3.14 while the project declares <3.14.

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