From 5acfae63f9dabdd36c6fda3e5ccb29323928c740 Mon Sep 17 00:00:00 2001 From: tqchen Date: Sat, 27 Mar 2021 08:41:02 -0400 Subject: [PATCH] fix lint --- python/tvm/exec/popen_worker.py | 2 +- python/tvm/exec/rpc_proxy.py | 6 +----- python/tvm/rpc/proxy.py | 2 +- 3 files changed, 3 insertions(+), 7 deletions(-) diff --git a/python/tvm/exec/popen_worker.py b/python/tvm/exec/popen_worker.py index 07c1d9c13e7c4..35571ac58de8c 100644 --- a/python/tvm/exec/popen_worker.py +++ b/python/tvm/exec/popen_worker.py @@ -22,8 +22,8 @@ import threading import traceback import pickle -import cloudpickle import logging +import cloudpickle from tvm.contrib.popen_pool import StatusKind diff --git a/python/tvm/exec/rpc_proxy.py b/python/tvm/exec/rpc_proxy.py index 9cf48406e88d2..bf315fdb90877 100644 --- a/python/tvm/exec/rpc_proxy.py +++ b/python/tvm/exec/rpc_proxy.py @@ -16,14 +16,10 @@ # under the License. # pylint: disable=redefined-outer-name, invalid-name """RPC web proxy, allows redirect to websocket based RPC servers(browsers)""" -from __future__ import absolute_import - import logging import argparse -import multiprocessing -import sys import os -from ..rpc.proxy import Proxy +from tvm.rpc.proxy import Proxy def find_example_resource(): diff --git a/python/tvm/rpc/proxy.py b/python/tvm/rpc/proxy.py index 8cabf3d7d0fb9..b66ad0e5d5aec 100644 --- a/python/tvm/rpc/proxy.py +++ b/python/tvm/rpc/proxy.py @@ -43,12 +43,12 @@ "RPCProxy module requires tornado package %s. Try 'pip install tornado'." % error_msg ) +from tvm.contrib.popen_pool import PopenWorker from . import _ffi_api from . import base from .base import TrackerCode from .server import _server_env from .._ffi.base import py_str -from tvm.contrib.popen_pool import PopenWorker class ForwardHandler(object):