From 6883443498b2c44a658ead84e63963f171979158 Mon Sep 17 00:00:00 2001 From: Jules van Rie Date: Thu, 23 May 2024 17:18:54 +0200 Subject: [PATCH] Relax pip check for tensorflow-cpu --- checks/pip_check.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/checks/pip_check.sh b/checks/pip_check.sh index 20b745c..26fb541 100644 --- a/checks/pip_check.sh +++ b/checks/pip_check.sh @@ -25,6 +25,11 @@ for r in ${REQUIRED[@]}; do if [ $r = ${q} ]; then present=1 fi + if [ $r = 'tensorflow' ]; then + if [ 'tensorflow-cpu' = $q ]; then + present=1 + fi + fi fi done if (($present == 0))