From ffffd94d2c5dc57081c838dd33b7fb4e699e52c2 Mon Sep 17 00:00:00 2001
From: Anirudh Dagar <anirudhdagar6@gmail.com>
Date: Wed, 9 Aug 2023 15:07:01 +0200
Subject: [PATCH] CI: Fix pytorch use newer numpy version

---
 .github/workflow_scripts/build_pytorch.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflow_scripts/build_pytorch.sh b/.github/workflow_scripts/build_pytorch.sh
index f52b5e428..d275ce04f 100644
--- a/.github/workflow_scripts/build_pytorch.sh
+++ b/.github/workflow_scripts/build_pytorch.sh
@@ -10,6 +10,7 @@ TARGET_BRANCH="$2" # Eg. 'master' ; if PR raised to master
 CACHE_DIR="$3"  # Eg. 'ci_cache_pr' or 'ci_cache_push'
 
 pip3 install d2l==0.17.6
+pip3 install --upgrade numpy  # numpy version installed by d2l==0.17.6 is too old
 mkdir _build
 
 source $(dirname "$0")/utils.sh