From 1ccca5a53d20c909f3ccf8a4b81df804e6717cb1 Mon Sep 17 00:00:00 2001
From: Wahab Kawafi <43846328+wahabk@users.noreply.github.com>
Date: Mon, 22 Jul 2024 16:00:19 +0100
Subject: [PATCH] Support `aarch64` with pip (#639)

* tensorflow-cpu -> tensorflow

* tf sys_platform and platform_machine
---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2e3a478d..e67828af 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,8 @@ absl-py = "^1.0.0"
 jax = { version = "^0.4.20", optional = true }
 matplotlib = "^3.2.2"
 tensorflow-macos = { version = "<2.14.0", markers = "sys_platform == 'darwin'" }
-tensorflow-cpu = { version = "^2.12.1", markers = "sys_platform != 'darwin'" }
+tensorflow-cpu = { version = "^2.12.1", markers = "sys_platform != 'darwin' and platform_machine != 'aarch64'" }
+tensorflow = { version = "^2.12.1", markers = "sys_platform != 'darwin' and platform_machine == 'aarch64'" }
 biopython = "<1.83"
 numpy = "^1.22.0"
 pandas = "^1.3.4"