def testIsInstanceNdarrayDuringTracing(self):
    arr = onp.ones(3)

    @api.jit
    def f(x):
        self.assertIsInstance(x, jnp.ndarray)
        return jnp.sum(x)
    f(arr)
def testIsInstanceNdarrayDuringTracing(self):
    arr = onp.ones(3)

    @api.jit
    def f(x):
        self.assertIsInstance(x, jnp.ndarray)
        return jnp.sum(x)
    f(arr)