From ba517f50d1d68d25fc96bd44b0d935175d685c56 Mon Sep 17 00:00:00 2001
From: chicm-ms <chec@microsoft.com>
Date: Wed, 18 Mar 2020 20:20:46 +0800
Subject: [PATCH] remove torch

---
 test/nni_test/nnitest/run_tests.py                  | 6 ------
 test/pipelines/pipelines-it-frameworkcontroller.yml | 1 -
 test/pipelines/pipelines-it-kubeflow.yml            | 1 -
 test/pipelines/pipelines-it-pai-windows.yml         | 1 -
 test/pipelines/pipelines-it-pai.yml                 | 1 -
 test/pipelines/pipelines-it-remote-windows.yml      | 1 -
 test/pipelines/pipelines-it-remote.yml              | 1 -
 7 files changed, 12 deletions(-)

diff --git a/test/nni_test/nnitest/run_tests.py b/test/nni_test/nnitest/run_tests.py
index 5c37dfe378..d387f814a2 100644
--- a/test/nni_test/nnitest/run_tests.py
+++ b/test/nni_test/nnitest/run_tests.py
@@ -9,7 +9,6 @@
 import shlex
 import traceback
 import json
-import torch
 import ruamel.yaml as yaml
 
 from utils import setup_experiment, get_experiment_status, get_yml_content, dump_yml_content, get_experiment_id, \
@@ -59,11 +58,6 @@ def run_test_case(test_case_config, it_config, args):
     if sys.platform == 'win32':
         test_yml_config['trial']['command'] = test_yml_config['trial']['command'].replace('python3', 'python')
 
-    # check GPU
-    if test_yml_config['trial']['gpuNum'] > 0 and torch.cuda.device_count() < 1:
-        print('skipping {}, gpu is not available'.format(test_case_config['name']))
-        return
-
     # generate temporary config yml file to launch experiment
     new_config_file = config_path + '.tmp'
     dump_yml_content(new_config_file, test_yml_config)
diff --git a/test/pipelines/pipelines-it-frameworkcontroller.yml b/test/pipelines/pipelines-it-frameworkcontroller.yml
index 57762f4585..9e21054202 100644
--- a/test/pipelines/pipelines-it-frameworkcontroller.yml
+++ b/test/pipelines/pipelines-it-frameworkcontroller.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB
diff --git a/test/pipelines/pipelines-it-kubeflow.yml b/test/pipelines/pipelines-it-kubeflow.yml
index a77c1274f4..e7c38861cc 100644
--- a/test/pipelines/pipelines-it-kubeflow.yml
+++ b/test/pipelines/pipelines-it-kubeflow.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB
diff --git a/test/pipelines/pipelines-it-pai-windows.yml b/test/pipelines/pipelines-it-pai-windows.yml
index 0617897989..6aa5a24a31 100644
--- a/test/pipelines/pipelines-it-pai-windows.yml
+++ b/test/pipelines/pipelines-it-pai-windows.yml
@@ -58,7 +58,6 @@ jobs:
   - script: |
       set PATH=$(ENV_PATH)
       python -m pip install scikit-learn==0.21.0 --user
-      python -m pip install torch===1.2.0 -f https://download.pytorch.org/whl/torch_stable.html --user
     displayName: 'Install dependencies for integration tests'
   - script: |
       cd test
diff --git a/test/pipelines/pipelines-it-pai.yml b/test/pipelines/pipelines-it-pai.yml
index 7f05c768fd..41c10e788d 100644
--- a/test/pipelines/pipelines-it-pai.yml
+++ b/test/pipelines/pipelines-it-pai.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB
diff --git a/test/pipelines/pipelines-it-remote-windows.yml b/test/pipelines/pipelines-it-remote-windows.yml
index e450a5c592..481a31353e 100644
--- a/test/pipelines/pipelines-it-remote-windows.yml
+++ b/test/pipelines/pipelines-it-remote-windows.yml
@@ -17,7 +17,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
   - script: |
       python -m pip install scikit-learn==0.20.1 --user
-      python -m pip install torch===1.2.0 -f https://download.pytorch.org/whl/torch_stable.html --user
     displayName: 'Install dependencies for integration tests'
   - task: SSH@0
     inputs:
diff --git a/test/pipelines/pipelines-it-remote.yml b/test/pipelines/pipelines-it-remote.yml
index 03165c1213..c8abae058f 100644
--- a/test/pipelines/pipelines-it-remote.yml
+++ b/test/pipelines/pipelines-it-remote.yml
@@ -15,7 +15,6 @@ jobs:
       source install.sh
     displayName: 'Install nni toolkit via source code'
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB