diff --git a/python/tvm/driver/tvmc/compiler.py b/python/tvm/driver/tvmc/compiler.py index 0cc26872cd47..071474a31594 100644 --- a/python/tvm/driver/tvmc/compiler.py +++ b/python/tvm/driver/tvmc/compiler.py @@ -38,7 +38,7 @@ @register_parser def add_compile_parser(subparsers): - """ Include parser for 'compile' subcommand """ + """Include parser for 'compile' subcommand""" parser = subparsers.add_parser("compile", help="compile a model.") parser.set_defaults(func=drive_compile) diff --git a/tests/lint/git-black.sh b/tests/lint/git-black.sh index 364f5bc93905..4052c052463e 100755 --- a/tests/lint/git-black.sh +++ b/tests/lint/git-black.sh @@ -37,6 +37,10 @@ if [[ "$#" -lt 1 ]]; then exit 1 fi +# required to make black's dep click to work +export LC_ALL=C.UTF-8 +export LANG=C.UTF-8 + if [ ! -x "$(command -v black)" ]; then echo "Cannot find black" exit 1 @@ -54,10 +58,6 @@ if [ -z ${FILES+x} ]; then fi echo "Files: $FILES" -# required to make black's dep click to work -export LC_ALL=C.UTF-8 -export LANG=C.UTF-8 - if [[ ${INPLACE_FORMAT} -eq 1 ]]; then echo "Running black on Python files against revision" $1: CMD=( "black" "${FILES[@]}" ) diff --git a/tests/python/frontend/tensorflow2/test_functional_models.py b/tests/python/frontend/tensorflow2/test_functional_models.py index 40d42a28025a..fe31a753746c 100644 --- a/tests/python/frontend/tensorflow2/test_functional_models.py +++ b/tests/python/frontend/tensorflow2/test_functional_models.py @@ -63,7 +63,7 @@ def run_model_graph(TestClass): def test_add_one(): class AddOne(tf.Module): - """ simple function to test x=x+1; scalar as input""" + """simple function to test x=x+1; scalar as input""" def get_input(self): return np.array(1.0, dtype="float32")