import sys
import importlib

sys.modules[__package__] = importlib.import_module('.dataset', __package__)

import os

__path__.append(os.path.dirname(__file__) + '/dataset')
from .dataset import *

Replace importlib hack with __path__
Replace importlib hack with __path__
