fix pipeline (#4721)

nni/algorithms/compression/tensorflow/pruning/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .one_shot_pruner import *

nni/algorithms/compression/tensorflow/pruning/one_shot_pruner.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 import tensorflow as tf
 
 from nni.compression.tensorflow import Pruner

nni/algorithms/compression/v2/pytorch/base/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .compressor import Compressor, LayerInfo
 from .pruner import Pruner, PrunerModuleWrapper
 from .scheduler import BasePruningScheduler, Task, TaskResult

nni/algorithms/compression/v2/pytorch/pruning/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .basic_pruner import *
 from .basic_scheduler import PruningScheduler
 from .iterative_pruner import *

nni/algorithms/compression/v2/pytorch/pruning/tools/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .base import (
     HookCollectorInfo,
     DataCollector,

nni/algorithms/compression/v2/pytorch/pruning/tools/rl_env/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .agent import DDPG
 from .amc_env import AMCEnv

nni/algorithms/compression/v2/pytorch/utils/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .config_validation import CompressorSchema
 from .pruning import (
     config_list_canonical,

nni/algorithms/feature_engineering/gbdt_selector/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .gbdt_selector import GBDTSelector
\ No newline at end of file

nni/algorithms/feature_engineering/gradient_selector/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .gradient_selector import FeatureGradientSelector
\ No newline at end of file

nni/compression/pytorch/quantization/literal.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from enum import Enum, EnumMeta
 
 

nni/compression/pytorch/quantization/observers.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from torch.quantization import default_weight_observer, default_histogram_observer
 from torch.quantization import RecordingObserver as _RecordingObserver
 

nni/compression/pytorch/quantization/settings.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from typing import Any, Optional
 
 from .literal import QuantDtype, QuantType, QuantScheme

nni/compression/pytorch/quantization/utils.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 import torch
 
 from nni.common.version import TORCH_VERSION

nni/compression/pytorch/quantization_speedup/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .integrated_tensorrt import CalibrateType, ModelSpeedupTensorRT
\ No newline at end of file

nni/compression/pytorch/speedup/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .compressor import ModelSpeedup
\ No newline at end of file

nni/compression/pytorch/utils/__init__.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 from .utils import *
 from .shape_dependency import *
 from .shape_dependency import ReshapeDependency

nni/compression/pytorch/utils/num_param_counter.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
 def get_total_num_weights(model, op_types=['default']):
         '''
         calculate the total number of weights

nni/experiment/launcher.py

@@ -120,7 +120,11 @@ def start_experiment(action, exp_id, config, port, debug, run_mode, url_prefix):
 
     link = Path(config.experiment_working_directory, '_latest')
     try:
-        link.unlink(missing_ok=True)
+        if sys.version_info >= (3, 8):
+            link.unlink(missing_ok=True)
+        else:
+            if link.exists():
+                link.unlink()
         link.symlink_to(exp_id, target_is_directory=True)
     except Exception:
         if sys.platform != 'win32':

setup_ts.py

@@ -223,7 +223,16 @@ def copy_nni_node(version):
     """
     _print('Copying files')
 
-    shutil.copytree('ts/nni_manager/dist', 'nni_node', dirs_exist_ok=True)
+    if sys.version_info >= (3, 8):
+        shutil.copytree('ts/nni_manager/dist', 'nni_node', dirs_exist_ok=True)
+    else:
+        for item in os.listdir('ts/nni_manager/dist'):
+            subsrc = os.path.join('ts/nni_manager/dist', item)
+            subdst = os.path.join('nni_node', item)
+            if os.path.isdir(subsrc):
+                shutil.copytree(subsrc, subdst)
+            else:
+                shutil.copy2(subsrc, subdst)
     shutil.copyfile('ts/nni_manager/yarn.lock', 'nni_node/yarn.lock')
     Path('nni_node/nni_manager.tsbuildinfo').unlink()