Add type.

apache · Sep 20, 2021 · 6099add · 6099add
1 parent b759069
commit 6099add
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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."