rl_code_fix_env / conftest.py
Viraj0112's picture
Upload folder using huggingface_hub
03a907a verified
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