Merge pull request #140 from Microsoft/master

Fix integration test dependencies (#822)

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

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