_aliases.py

import sys
import importlib
from pathlib import Path

_REPO_ROOT = str(Path(__file__).parent)
if _REPO_ROOT not in sys.path:
    sys.path.insert(0, _REPO_ROOT)

import dataset as _real_dataset

sys.modules.setdefault("src.dataset", _real_dataset)

import pkgutil
for _pkg in pkgutil.iter_modules(_real_dataset.__path__):
    _full = f"dataset.{_pkg.name}"
    _alias = f"src.dataset.{_pkg.name}"
    try:
        _mod = importlib.import_module(_full)
        sys.modules.setdefault(_alias, _mod)
    except Exception:
        pass