fix: freeze constant update calibration after one probe

1 files changed, 8 insertions, 2 deletions

diff --git a/experiments/rain_ep_dillavou_smoke.py b/experiments/rain_ep_dillavou_smoke.py
index d4a634a..9519210 100644
--- a/experiments/rain_ep_dillavou_smoke.py
+++ b/experiments/rain_ep_dillavou_smoke.py
@@ -70,9 +70,11 @@ def main() -> None:
     assert fixed_relative_error < 2e-6, fixed_relative_error
 
     constant = DillavouUpdateCorrector(
-        mode="constant", bias_ratio=0.2, predictor_rate=1.0, seed=41)
+        mode="constant", bias_ratio=0.2, predictor_rate=1.0,
+        calibration_steps=1, neutral_cadence=0, seed=41)
     innovation = DillavouUpdateCorrector(
-        mode="innovation", bias_ratio=0.2, predictor_rate=1.0, seed=41)
+        mode="innovation", bias_ratio=0.2, predictor_rate=1.0,
+        calibration_steps=1, neutral_cadence=0, seed=41)
     corrected_constant = constant.apply(clean_a, parameters_a)
     corrected_innovation = innovation.apply(clean_a, parameters_a)
     constant_error = max(
@@ -85,6 +87,10 @@ def main() -> None:
     )
     assert constant_error < 2e-7, constant_error
     assert innovation_error < 2e-7, innovation_error
+    constant.apply(clean_b, parameters_b)
+    innovation.apply(clean_b, parameters_b)
+    assert constant.debiaser.neutral_observations == 1
+    assert innovation.debiaser.neutral_observations == 1
 
     # Integration check: the corruption is attached after Rain's hand-written
     # local EP estimator and introduces no autograd graph.