search

qutip / qutip-jax

JAX backend for QuTiP
BSD 3-Clause "New" or "Revised" License
18 stars 7 forks source link

iszero_jaxarray needs a default value for tol #17

Closed BoxiLi closed 1 year ago

BoxiLi commented 1 year ago

The rendering of qutip matrix in IPython fails

qutip.sigmax().to("jax").__str__()

with the error 

TypeError: iszero_jaxarray() missing 1 required positional argument: 'tol'

I think the definition of iszero_jaxarray needs a default value for tol, e.g.

iszero_jaxarray(matrix, tol=None)