From 9738293a25fc0bd8074e0a7c5be1746b23cff979 Mon Sep 17 00:00:00 2001
From: Felix Chern <fchern@google.com>
Date: Thu, 16 Oct 2025 08:21:34 -0700
Subject: [PATCH] Fix build_whl

PiperOrigin-RevId: 820243641
---
 oss/Dockerfile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/oss/Dockerfile b/oss/Dockerfile
index 095f03e..762ceee 100644
--- a/oss/Dockerfile
+++ b/oss/Dockerfile
@@ -21,10 +21,13 @@ RUN BAZEL_ARCH_SUFFIX="$(uname -m | sed s/aarch64/arm64/)" \
   && curl -sSL --fail -o /usr/local/bin/bazel "https://github.com/bazelbuild/bazel/releases/download/$BAZEL_VERSION/bazel-$BAZEL_VERSION-linux-$BAZEL_ARCH_SUFFIX" \
   && chmod a+x /usr/local/bin/bazel
 
+RUN apt-get install -y libhdf5-dev hdf5-tools
+
 # Install dependencies needed for array_record.
 RUN --mount=type=cache,target=/root/.cache \
   $PYTHON_BIN -m pip install -U \
     absl-py \
+    h5py \
     auditwheel \
     etils[epath] \
     patchelf \
@@ -32,4 +35,4 @@ RUN --mount=type=cache,target=/root/.cache \
     twine \
     wheel;
 
-WORKDIR "/tmp/array_record"
\ No newline at end of file
+WORKDIR "/tmp/array_record"