From d262ef97cdf72f4c9efd4b0b3ad27b1ff595dc68 Mon Sep 17 00:00:00 2001
From: Marvin Ritter <marvinritter@google.com>
Date: Fri, 6 Oct 2023 08:19:06 -0700
Subject: [PATCH] Make new wheel speicify they are arch dependent.

This is manually merging #79 since our current setup doesn't support merging external PRs.

Also bump version to 0.4.2.

PiperOrigin-RevId: 571342678
---
 setup.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 0328580..afdfea7 100644
--- a/setup.py
+++ b/setup.py
@@ -2,15 +2,24 @@
 
 from setuptools import find_packages
 from setuptools import setup
+from setuptools.dist import Distribution
 
 REQUIRED_PACKAGES = [
     'absl-py',
     'etils[epath]',
 ]
 
+
+class BinaryDistribution(Distribution):
+  """This class makes 'bdist_wheel' include an ABI tag on the wheel."""
+
+  def has_ext_modules(self):
+    return True
+
+
 setup(
     name='array_record',
-    version='0.4.1',
+    version='0.4.2',
     description='A file format that achieves a new frontier of IO efficiency',
     author='ArrayRecord team',
     author_email='no-reply@google.com',
@@ -27,4 +36,5 @@
         'Programming Language :: Python :: 3.11',
     ],
     zip_safe=False,
+    distclass=BinaryDistribution,
 )