diff --git a/python/tvm/meta_schedule/tune_context.py b/python/tvm/meta_schedule/tune_context.py index f0e92bed17019..eec242e13c4b4 100644 --- a/python/tvm/meta_schedule/tune_context.py +++ b/python/tvm/meta_schedule/tune_context.py @@ -92,7 +92,7 @@ def __init__( num_threads = cpu_count() self.__init_handle_by_constructor__( - _ffi_api.TuneContext, # pylint: disable=no-member + _ffi_api.TuneContext, # type: ignore # pylint: disable=no-member mod, target, task_name, diff --git a/tests/scripts/task_mypy.sh b/tests/scripts/task_mypy.sh index 8507f311e9da7..05d1c238b64f2 100755 --- a/tests/scripts/task_mypy.sh +++ b/tests/scripts/task_mypy.sh @@ -17,13 +17,16 @@ # under the License. set -o pipefail -echo "Checking MyPy Type defs in the schedule package." +echo "Checking MyPy Type defs in the TensorIR schedule package." mypy --check-untyped-defs python/tvm/tir/schedule +echo "Checking MyPy Type defs in the meta schedule package." +mypy --check-untyped-defs python/tvm/meta_schedule + echo "Checking MyPy Type defs in the analysis package." mypy --check-untyped-defs python/tvm/tir/analysis/ -echo "Checking MyPy Type defs in the transofrm package." +echo "Checking MyPy Type defs in the transform package." mypy --check-untyped-defs python/tvm/tir/transform/ echo "Checking MyPy Type defs in the tvm.relay.backend.contrib.ethosu package."