Fix integration test dependencies (#822)

* Install dependencies for PAI/k8s IT * updates
microsoft · Mar 11, 2019 · 9790e97 · 9790e97
1 parent 7108466
commit 9790e97
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/test/pipelines-it-kubeflow.yml b/test/pipelines-it-kubeflow.yml
@@ -38,6 +38,16 @@ jobs:
       source install.sh
     displayName: 'Install nni toolkit via source code'
 
+  - script: |
+      python3 -m pip install scikit-learn==0.20.0 --user
+      python3 -m pip install torch==0.4.1 --user
+      python3 -m pip install torchvision==0.2.1 --user
+      python3 -m pip install keras==2.1.6 --user
+      python3 -m pip install tensorflow-gpu==1.10.0 --user
+      sudo apt-get install swig -y
+      nnictl package install --name=SMAC
+    displayName: 'Install dependencies for integration tests in Kubeflow mode'
+
   - script: |
       if [ $(build_docker_img) = 'true' ]
       then

diff --git a/test/pipelines-it-pai.yml b/test/pipelines-it-pai.yml
@@ -38,6 +38,16 @@ jobs:
       source install.sh
     displayName: 'Install nni toolkit via source code'
 
+  - script: |
+      python3 -m pip install scikit-learn==0.20.0 --user
+      python3 -m pip install torch==0.4.1 --user
+      python3 -m pip install torchvision==0.2.1 --user
+      python3 -m pip install keras==2.1.6 --user
+      python3 -m pip install tensorflow-gpu==1.10.0 --user
+      sudo apt-get install swig -y
+      nnictl package install --name=SMAC
+    displayName: 'Install dependencies for integration tests in PAI mode'
+
   - script: |
       if [ $(build_docker_img) = 'true' ]
       then