def wrapped2(): wrapped() with option_context({'eager_mode': True}): assert options.eager_mode assert not utils.is_eager_mode() with utils.enter_mode(kernel=False): assert not utils.is_kernel_mode() assert utils.is_kernel_mode()
def wrapped2(): wrapped() with option_context({'eager_mode': True}): self.assertTrue(options.eager_mode) self.assertFalse(utils.is_eager_mode()) with utils.enter_mode(kernel=False): self.assertFalse(utils.is_kernel_mode()) self.assertTrue(utils.is_kernel_mode())
def wrapped(): return utils.is_eager_mode()