From 039af4ecdeae7e655455de3f4324a8a2adbf5d3b Mon Sep 17 00:00:00 2001
From: gmm <38800877+mmglove@users.noreply.github.com>
Date: Wed, 19 Jun 2024 15:46:12 +0800
Subject: [PATCH] Update prepare.sh

---
 tests/test_tipc/prepare.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_tipc/prepare.sh b/tests/test_tipc/prepare.sh
index eb17ed49a..7d4dd8b16 100755
--- a/tests/test_tipc/prepare.sh
+++ b/tests/test_tipc/prepare.sh
@@ -35,7 +35,8 @@ if [[ ${MODE} = "benchmark_train" ]];then
     pip install setuptools_scm #-i https://pypi.tuna.tsinghua.edu.cn/simple 
     pip install . #-i https://pypi.tuna.tsinghua.edu.cn/simple 
     pip install jsonlines
-    pip install scipy==1.12.0
+    pip install -U scipy==1.12.0  # 高版本数据处理部分报错
+    pip install -U matplotlib==3.7.1 # 高版本报错cannot import name 'get_cmap' from 'matplotlib.cm' 
     pip list
     cd -
     if [[ ${model_name} == "conformer" ]]; then