src/utils/__init__.py

"""Utility functions for the audit assistant."""

from .device import (
    get_device,
    get_device_for_sentence_transformers,
    get_device_for_colpali,
    log_device_info,
    get_cached_device
)

__all__ = [
    "get_device",
    "get_device_for_sentence_transformers",
    "get_device_for_colpali",
    "log_device_info",
    "get_cached_device"
]