Don't install tensorflow-text on M1 OSX.

PiperOrigin-RevId: 617139667
Change-Id: I6ce040d143c312550c7983f0f50c4f334499ad93

setup.py

@@ -15,8 +15,33 @@
 
 """Install script for setuptools."""
 
+import platform
 import setuptools
 
+IS_M1_OSX = platform.system() == 'Darwin' and platform.machine() == 'arm64'
+
+REQUIREMENTS = (
+    # TODO: b/312199199 - remove some requirements.
+    'absl-py',
+    'google-cloud-aiplatform',
+    'ipython',
+    'matplotlib',
+    'numpy',
+    'openai>=1.3.0',
+    'pandas<=2.0.3',
+    'python-dateutil',
+    'reactivex',
+    'retry',
+    'scipy',
+    'tensorflow',
+    'tensorflow-hub',
+    'tensorflow-text',
+    'termcolor',
+    'typing-extensions',
+)
+M1_OSX_REQUIREMENTS = tuple(set(REQUIREMENTS) - {'tensorflow-text'})
+
+
 setuptools.setup(
     name='gdm-concordia',
     version='1.2.0',
@@ -49,25 +74,7 @@
     packages=setuptools.find_packages(include=['concordia', 'concordia.*']),
     package_data={},
     python_requires='>=3.11',
-    install_requires=[
-        # TODO: b/312199199 - remove some requirements.
-        'absl-py',
-        'google-cloud-aiplatform',
-        'ipython',
-        'matplotlib',
-        'numpy',
-        'openai>=1.3.0',
-        'pandas<=2.0.3',
-        'python-dateutil',
-        'reactivex',
-        'retry',
-        'scipy',
-        'tensorflow',
-        'tensorflow-hub',
-        'tensorflow-text',
-        'termcolor',
-        'typing-extensions',
-    ],
+    install_requires=M1_OSX_REQUIREMENTS if IS_M1_OSX else REQUIREMENTS,
     extras_require={
         # Used in development.
         'dev': [