Relax pip check for tensorflow-cpu

lewagon · May 23, 2024 · 6883443 · 6883443
1 parent 8518a99
commit 6883443
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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))