diff --git a/sdk/load-and-run/dump_with_testcase_mge.py b/sdk/load-and-run/dump_with_testcase_mge.py index cec42f85..8180a7f9 100755 --- a/sdk/load-and-run/dump_with_testcase_mge.py +++ b/sdk/load-and-run/dump_with_testcase_mge.py @@ -14,7 +14,7 @@ import struct import cv2 import numpy as np -import megbrain as mgb +import megengine._internal as mgb import megengine as mge logger = mge.get_logger(__name__)