From de3317e6c758d8e77b98b351d59272c1c97e5060 Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Thu, 8 Jul 2021 11:24:43 -0500 Subject: [PATCH] [PyLint] Minor updates to pass pylint locally. With either the ci_lint docker image, or the matched version of pylint==2.4.4, I got two lint errors running locally that didn't show up in the CI. Fixing them. --- python/tvm/script/diagnostics.py | 3 ++- python/tvm/topi/random/kernel.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/python/tvm/script/diagnostics.py b/python/tvm/script/diagnostics.py index fc196f6b16ae..e676461ab39e 100644 --- a/python/tvm/script/diagnostics.py +++ b/python/tvm/script/diagnostics.py @@ -17,8 +17,9 @@ """Bridge from synr's (the library used for parsing the python AST) DiagnosticContext to TVM's diagnostics """ -import tvm from synr import DiagnosticContext, ast + +import tvm from tvm.ir.diagnostics import DiagnosticContext as TVMCtx from tvm.ir.diagnostics import get_renderer, DiagnosticLevel, Diagnostic diff --git a/python/tvm/topi/random/kernel.py b/python/tvm/topi/random/kernel.py index 8b6bb114b181..2ef97e2edc5c 100644 --- a/python/tvm/topi/random/kernel.py +++ b/python/tvm/topi/random/kernel.py @@ -15,9 +15,11 @@ # specific language governing permissions and limitations # under the License. """Pseudorandom number kernels.""" +import numpy as np + import tvm import tvm.topi -import numpy as np + from ... import tir from ...tir import ir_builder