import os
import sys

from omegaconf import OmegaConf

from larm.common.registry import registry

from larm.data.builders import *
from larm.memory_generator import *


root_dir = os.path.dirname(os.path.abspath(__file__))
default_cfg = OmegaConf.load(os.path.join(root_dir, "configs/default.yaml"))


registry.register_path("library_root", root_dir)  
repo_root = os.path.join(root_dir, "..")
registry.register_path("repo_root", repo_root)     
cache_root = os.path.join(repo_root, default_cfg.env.cache_root)
registry.register_path("cache_root", cache_root)  
results_root = os.path.join(repo_root, default_cfg.env.results_root)
registry.register_path("results_root", results_root)