def testTracingPrimitiveWithNoTranslationErrorMessage(self):
    self.skipTest('test needs jax3 update')
    foo = jnp._not_implemented((lambda x: x))
    foo(onp.arange(3))
    cfoo = api.jit(foo)
    self.assertRaises(NotImplementedError, (lambda : cfoo(onp.arange(3))))
def testTracingPrimitiveWithNoTranslationErrorMessage(self):
    self.skipTest('test needs jax3 update')
    foo = jnp._not_implemented((lambda x: x))
    foo(onp.arange(3))
    cfoo = api.jit(foo)
    self.assertRaises(NotImplementedError, (lambda : cfoo(onp.arange(3))))