Function _convert_to_numpy

sklearn/utils/_array_api.py:999–1018

Convert X into a NumPy ndarray on the CPU. This function uses library-specific methods to convert the array to a NumPy ndarray on the CPU. It is only meant as a fallback when move_to fails to use the DLPACK protocol. This function is not meant to be called directly and `move_to

(array, xp)

def _convert_to_numpy(array, xp):
    """Convert X into a NumPy ndarray on the CPU.

    This function uses library-specific methods to convert the array to a NumPy
    ndarray on the CPU. It is only meant as a fallback when move_to fails to use the
    DLPACK protocol.

    This function is not meant to be called directly and
    `move_to(array, xp=np, device="cpu")` should be used instead.
    """
    if _is_xp_namespace(xp, "torch"):
        return array.cpu().numpy()
    elif _is_xp_namespace(xp, "cupy"):  # pragma: nocover
        return array.get()
    elif _is_xp_namespace(xp, "array_api_strict"):
        return numpy.asarray(xp.asarray(array, device=xp.Device("CPU_DEVICE")))
    elif _is_xp_namespace(xp, "dpnp"):  # pragma: nocover
        return array.asnumpy()

    return numpy.asarray(array)


def _estimator_with_converted_arrays(estimator, converter):