Fix build_whl

PiperOrigin-RevId: 820243641

Author: dryman

oss/Dockerfile

@@ -21,15 +21,18 @@ 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 \
     setuptools \
     twine \
     wheel;
 
-WORKDIR "/tmp/array_record"
+WORKDIR "/tmp/array_record"