From a4fc47f5ca66ea56b645fff46e08d300b574cfb1 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Thu, 6 Aug 2026 08:36:19 -0700
Subject: [PATCH] Fix int64 multiplication overflow in PyGrain mixture index
 dataset sampler when using small dataset weights over large numbers of
 records.

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

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 63df64230..2d1ec04e0 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -17,6 +17,8 @@ changes. Best viewed [here](https://google-grain.readthedocs.io/en/latest/change
 * Deprecations:
 
 * Bug fixes:
+  * Fixed int64 multiplication overflow in mixture index dataset sampler when
+    using small dataset weights over large numbers of records.
   * Fixed bug in DataLoader where sharding remainder was dropped even when ShardOptions.drop_remainder=False.
   * Fixes reference cycle in BatchMapDataset.