From ad6bb1e7d960c636ba3f21fd05cdffb09b916217 Mon Sep 17 00:00:00 2001
From: "mulin.lyh" <mulin.lyh@taobao.com>
Date: Wed, 7 Sep 2022 20:51:15 +0800
Subject: [PATCH] [to #44790143]fix: add ipythonkernel to image for dsw        
 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10049527

    * add ipykernel to image for dsw
---
 docker/Dockerfile.ubuntu | 4 +++-
 tests/run.py             | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu
index 78da0b6f..e0bfa908 100644
--- a/docker/Dockerfile.ubuntu
+++ b/docker/Dockerfile.ubuntu
@@ -75,7 +75,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
 ENV SHELL=/bin/bash
 
 # install special package
-RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 numpy==1.18.5 datasets==2.1.0
+RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 datasets==2.1.0 ipykernel && \
+    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
+    pip config set install.trusted-host pypi.tuna.tsinghua.edu.cn
 
 RUN if [ "$USE_GPU" = "True" ] ; then \
         pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \
diff --git a/tests/run.py b/tests/run.py
index 51a563fe..18839622 100644
--- a/tests/run.py
+++ b/tests/run.py
@@ -420,7 +420,7 @@ if __name__ == '__main__':
     parser.add_argument(
         '--suites',
         nargs='*',
-        help='Run specified test suites(test suite file list)')
+        help='Run specified test suites(test suite files list split by space)')
     args = parser.parse_args()
     set_test_level(args.level)
     os.environ['REGRESSION_BASELINE'] = '1'