fix wrong arg call

This commit is contained in:
shijianning 2020-05-13 16:21:24 +08:00
parent 5afcefdce4
commit 2cf2a67dfd
1 changed files with 2 additions and 1 deletions

View File

@ -31,8 +31,9 @@ def AKGAddPath():
class AKGMetaPathFinder:
"""class AKGMetaPath finder."""
def find_module(self, fullname):
def find_module(self, fullname, path=None):
"""method _akg find module."""
_ = path
if fullname.startswith("_akg.tvm"):
rname = fullname[5:]
return AKGMetaPathLoader(rname)