add type annotation to the return value

apache · Aug 19, 2022 · 2575d7c · 2575d7c
1 parent 959a00b
commit 2575d7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tir/schedule/trace.cc b/src/tir/schedule/trace.cc
@@ -477,7 +477,7 @@ TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
       const auto* self = obj.as<TraceNode>();
       ICHECK_NOTNULL(self);
       p->stream << "# from tvm import tir" << std::endl;
-      p->stream << "def apply_trace(sch: tir.Schedule):" << std::endl;
+      p->stream << "def apply_trace(sch: tir.Schedule) -> None:" << std::endl;
       Array<String> repr = self->AsPython(/*remove_postproc=*/false);
       bool is_first = true;
       for (const String& line : repr) {