diff --git a/sdk/load-and-run/dump_with_testcase_mge.py b/sdk/load-and-run/dump_with_testcase_mge.py index cec42f85e0c4412acaab68708d4a7c63c816c67c..8180a7f974e92d8dd7a01bd7d8120fd78c961318 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__)