From 120655b75a63a4930bdc306d1646f51fff8951ab Mon Sep 17 00:00:00 2001
From: Frederic Bastien <fbastien@nvidia.com>
Date: Tue, 31 Oct 2023 07:18:57 -0700
Subject: [PATCH] WAR bad array_record wheel on arm64

---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b97a158ae10..ec3820b9ff8 100644
--- a/setup.py
+++ b/setup.py
@@ -60,7 +60,10 @@
 
 REQUIRED_PKGS = [
     'absl-py',
-    'array_record',
+# Min version of 0.5.0 as old array_record wheel are bugged on all
+# platform except 'x86_64'. See
+# https://github.com/google/array_record/issues/71
+    'array_record>=0.5.0',
     'click',
     'dm-tree',
     'etils[enp,epath,etree]>=0.9.0',