Changing ks2d2s to estat2d raises an error

[skoldobskiy]

Hello!

Thanks a lot for python implementation of 2D KS test, it is really helpful!
I also tried to use estat2d, but seems it does not work even for the example given on main page.
Changing ks2d2s to estat2d:

# generate mock samples for the test
np.random.seed(42)
x1, y1 = np.random.randn(2, 100)
x2, y2 = np.random.randn(2, 100)  # same distribution as (x1, y1)

p,en,en_boot = estat2d(x1, y1, x2, y2)
print(f"{p=:.3g}")

raises an error:

    p = (en_boot >= en).sum() / nboot

TypeError: '>=' not supported between instances of 'float' and 'NoneType'

UPD: It was wrong from my side. The function is actually working.