Function get_torch

sdk/python/feast/torch_wrapper.py:22–39

Return the torch module if available, else raise a friendly error. This prevents crashing on import if CUDA libs are missing.

()

def get_torch():
    """
    Return the torch module if available, else raise a friendly error.

    This prevents crashing on import if CUDA libs are missing.
    """
    if TORCH_AVAILABLE:
        return _torch
    else:
        error_message = (
            "Torch is not available or failed to import.\n"
            "Original error:\n"
            f"{_torch_import_error}\n\n"
            "If you are on a CPU-only system, make sure you install the CPU-only torch wheel:\n"
            "  pip install torch torchvision -f https://download.pytorch.org/whl/cpu\n"
            "Or check your CUDA installation if using GPU torch.\n"
        )
        raise ImportError(error_message) from _torch_import_error