How to use binary_search overloads in python.

[LeonardEyer]

Basically, I want to do this kind of computation:

x = Float.linspace(0, 1, 100)

def f(i : UInt32) -> Mask:
    return ek.gather(x, i) > 0.5

ek.binary_search(0, 100, f)

With Float and UInt32 being cuda arrays.
But I get the following error: Unable to cast Python instance of type <class 'enoki.cuda_autodiff.Mask'> to C++ type 'bool'
It seems like pybind doesn't choose the correct overloaded method. Am I missing something?