utils/__init__.py

from utils.backend import Backend, detect, spaces_gpu_or_noop
from utils.budget import MODE_BUDGET, duration_for, size_for

__all__ = [
    "Backend",
    "detect",
    "spaces_gpu_or_noop",
    "MODE_BUDGET",
    "duration_for",
    "size_for",
]