Fix flake8 scaned python errors (#1241)

microsoft · Jul 10, 2019 · 32e4960 · 32e4960
1 parent 26f6f07
commit 32e4960
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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
@@ -20,6 +20,7 @@
 
 
 import random
+import numpy as np
 
 from .env_vars import trial_env_vars
 from . import trial