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 @@
import os
import sys
import time
import math
import torch
import torch.nn as nn
import torch.nn.init as init
......
......@@ -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)
......
......@@ -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):
......
......@@ -20,6 +20,7 @@
import random
import numpy as np
from .env_vars import trial_env_vars
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