def testClipError(self):
    with self.assertRaisesRegex(ValueError, 'At most one of a_min and a_max.*'):
        jnp.clip(jnp.zeros((3,)))
def testClipError(self):
    with self.assertRaisesRegex(ValueError, 'At most one of a_min and a_max.*'):
        jnp.clip(jnp.zeros((3,)))