From 32e496007c746c5d06c41952866050129a636160 Mon Sep 17 00:00:00 2001
From: chicm-ms <38930155+chicm-ms@users.noreply.github.com>
Date: Wed, 10 Jul 2019 15:59:49 +0800
Subject: [PATCH] Fix flake8 scaned python errors (#1241)

---
 examples/trials/cifar10_pytorch/utils.py        | 2 +-
 examples/trials/kaggle-tgs-salt/predict.py      | 2 +-
 src/sdk/pynni/nni/curvefitting_assessor/test.py | 2 ++
 src/sdk/pynni/nni/smartparam.py                 | 1 +
 4 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/examples/trials/cifar10_pytorch/utils.py b/examples/trials/cifar10_pytorch/utils.py
index 9d559bc23b..65e0d086b8 100644
--- a/examples/trials/cifar10_pytorch/utils.py
+++ b/examples/trials/cifar10_pytorch/utils.py
@@ -6,8 +6,8 @@
 import os
 import sys
 import time
-import math
 
+import torch
 import torch.nn as nn
 import torch.nn.init as init
 
diff --git a/examples/trials/kaggle-tgs-salt/predict.py b/examples/trials/kaggle-tgs-salt/predict.py
index 29f3da5409..2780d25708 100644
--- a/examples/trials/kaggle-tgs-salt/predict.py
+++ b/examples/trials/kaggle-tgs-salt/predict.py
@@ -133,7 +133,7 @@ def generate_preds(outputs, target_size, pad_mode, threshold=0.5):
         if pad_mode == 'resize':
             cropped = resize_image(output, target_size=target_size)
         else:
-            cropped = crop_image_softmax(output, target_size=target_size)
+            cropped = crop_image(output, target_size=target_size)
         pred = binarize(cropped, threshold)
         preds.append(pred)
 
diff --git a/src/sdk/pynni/nni/curvefitting_assessor/test.py b/src/sdk/pynni/nni/curvefitting_assessor/test.py
index bcbc1a0a6f..14601ccf1c 100644
--- a/src/sdk/pynni/nni/curvefitting_assessor/test.py
+++ b/src/sdk/pynni/nni/curvefitting_assessor/test.py
@@ -15,9 +15,11 @@
 # DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
+import numpy as np
 import unittest
 
 from .curvefitting_assessor import CurvefittingAssessor
+from .model_factory import CurveModel
 from nni.assessor import AssessResult
 
 class TestCurveFittingAssessor(unittest.TestCase):
diff --git a/src/sdk/pynni/nni/smartparam.py b/src/sdk/pynni/nni/smartparam.py
index 07519b69ea..b6d87d04e6 100644
--- a/src/sdk/pynni/nni/smartparam.py
+++ b/src/sdk/pynni/nni/smartparam.py
@@ -20,6 +20,7 @@
 
 
 import random
+import numpy as np
 
 from .env_vars import trial_env_vars
 from . import trial