[Bugfix] BuilderInput with default params (#532)

tlc-pack · Dec 3, 2021 · 3b0983e · 3b0983e
1 parent ab10735
commit 3b0983e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/tvm/meta_schedule/builder/builder.py b/python/tvm/meta_schedule/builder/builder.py
@@ -45,7 +45,12 @@ class BuilderInput(Object):
     target: Target
     params: Optional[Dict[str, NDArray]]
 
-    def __init__(self, mod: IRModule, target: Target, params: Optional[Dict[str, NDArray]]) -> None:
+    def __init__(
+        self,
+        mod: IRModule,
+        target: Target,
+        params: Optional[Dict[str, NDArray]] = None,
+    ) -> None:
         """Constructor.
 
         Parameters