From 02dddfa25fcefc61c7d61eaa3e5bef716d27d113 Mon Sep 17 00:00:00 2001
From: Tom Bocklisch <tom@rasa.com>
Date: Mon, 12 Sep 2022 21:37:01 +0200
Subject: [PATCH] install tensorflow-metal in arm linux container

---
 pyproject.toml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2f3615d02325..089a81ff7d0a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -120,9 +120,9 @@ sklearn-crfsuite = "~0.3"
 # https://europe-west3-python.pkg.dev/rasa-releases/psycopg-binary/
 psycopg2-binary = ">=2.8.2,<2.10.0"
 python-dateutil = "~2.8"
-tensorflow = { version = "~2.8.2", markers="sys_platform != 'darwin' or platform_machine != 'arm64'"} 
-tensorflow-macos = { version = "2.8.0", markers="sys_platform == 'darwin' and platform_machine == 'arm64'"}
-tensorflow-metal = { version = "0.5.1", markers="sys_platform == 'darwin' and platform_machine == 'arm64'"}
+tensorflow = { version = "~2.8.2", markers="platform_machine != 'arm64'"} 
+tensorflow-macos = { version = "2.8.0", markers="platform_machine == 'arm64'"}
+tensorflow-metal = { version = "0.5.1", markers="platform_machine == 'arm64'"}
 # protobuf in newer versions currenlty creates issues for tensorflow
 # see: https://github.com/tensorflow/tensorflow/blob/v2.8.2/tensorflow/tools/pip_package/setup.py#L84
 protobuf = ">=3.9.2,< 3.20"