Unverified Commit 32e49600 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix flake8 scaned python errors (#1241)

parent 26f6f07f
...@@ -6,8 +6,8 @@ ...@@ -6,8 +6,8 @@
import os import os
import sys import sys
import time import time
import math
import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.init as init import torch.nn.init as init
......
...@@ -133,7 +133,7 @@ def generate_preds(outputs, target_size, pad_mode, threshold=0.5): ...@@ -133,7 +133,7 @@ def generate_preds(outputs, target_size, pad_mode, threshold=0.5):
if pad_mode == 'resize': if pad_mode == 'resize':
cropped = resize_image(output, target_size=target_size) cropped = resize_image(output, target_size=target_size)
else: else:
cropped = crop_image_softmax(output, target_size=target_size) cropped = crop_image(output, target_size=target_size)
pred = binarize(cropped, threshold) pred = binarize(cropped, threshold)
preds.append(pred) preds.append(pred)
......
...@@ -15,9 +15,11 @@ ...@@ -15,9 +15,11 @@
# DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # 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. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
import numpy as np
import unittest import unittest
from .curvefitting_assessor import CurvefittingAssessor from .curvefitting_assessor import CurvefittingAssessor
from .model_factory import CurveModel
from nni.assessor import AssessResult from nni.assessor import AssessResult
class TestCurveFittingAssessor(unittest.TestCase): class TestCurveFittingAssessor(unittest.TestCase):
......
...@@ -20,6 +20,7 @@ ...@@ -20,6 +20,7 @@
import random import random
import numpy as np
from .env_vars import trial_env_vars from .env_vars import trial_env_vars
from . import trial from . import trial
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment