From 49c59d8b4192a5f4c9fcbb881d2496e3086d3cda Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Tue, 19 May 2026 22:35:56 -0700
Subject: [PATCH] Update TfGrainMitxure to provide batch size to
 TfMixtureIndexSampler

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

diff --git a/CHANGELOG.md b/CHANGELOG.md
index ec08be82c..04bb79c11 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,7 +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 accept dynamic Tensor/Variable proportions
-    that could be updated during runtime.
+    that could be updated during runtime. The proportions can support rank of 1
+    and 2.
   * 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.