From f4d1c367ab28f93efea19e8d7c80678aa593d8fe Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Mon, 11 May 2026 12:28:12 -0700
Subject: [PATCH] Updates TfMixtureIndexSampler so that proportions could be
 updated during iteration.

PiperOrigin-RevId: 913827623
---
 CHANGELOG.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index bfa0f5c6d..c0ba3efdc 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,8 @@ changes. Best viewed [here](https://google-grain.readthedocs.io/en/latest/change
   * Adds experimental `ShapeDtypeStructProtocol` and `ShapeDtypeStruct` to
     represent dataset element specs.
   * Updates TfMixtureIndexSampler to support datasets with weights of 0.
+  * Updates TfMixtureIndexSampler so that proportions could be updated during
+    iteration.
   * Adds profiling of multiprocess workers when using XProf profiler. To enable,
     set flag `grain_enable_multiprocess_worker_profiling=true` and add
     `"profile_subprocesses" = True` in advanced profiler options.