def f(pred): assert python_should_be_running checkify.assert_(pred, "foo")
def f(): checkify.assert_(False, "hi")
def f(x): checkify.assert_(x > 0, "must be positive!") return jnp.log(x)
def multi_errors(x): x = x / 0 # DIV x = jnp.sin(x) # NAN x = x[500] # OOB checkify.assert_(x < 0, "must be negative!") # ASSERT return x