model/pxdesign/__init__.py

import os
from pathlib import Path

parent_dir = Path(__file__).resolve().parent.parent

if os.environ.get("PROTENIX_DATA_ROOT_DIR") is None:
    os.environ["PROTENIX_DATA_ROOT_DIR"] = str(
        parent_dir / "release_data" / "ccd_cache"
    )

if os.environ.get("TOOL_WEIGHTS_ROOT") is None:
    os.environ["TOOL_WEIGHTS_ROOT"] = str(parent_dir / "tool_weights")