From fc7c9e79ca958ca63b4d66d52156e1f996639735 Mon Sep 17 00:00:00 2001
From: Younghee Kwon <youngheek@google.com>
Date: Sun, 17 May 2026 14:33:32 -0700
Subject: [PATCH] bugfix in stats module.

PiperOrigin-RevId: 916900498
---
 grain/_src/python/dataset/stats.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/grain/_src/python/dataset/stats.py b/grain/_src/python/dataset/stats.py
index d350399c4..03b45fb03 100644
--- a/grain/_src/python/dataset/stats.py
+++ b/grain/_src/python/dataset/stats.py
@@ -22,6 +22,7 @@
 import dataclasses
 import enum
 import functools
+import html
 from multiprocessing import queues
 import pprint
 import queue