"""KEV package defaults.""" import os # Set before submodules import PyTorch; callers can override this before importing KEV. os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")