FIX Allow multi-output in KernelRidge