def test_on_forward(self): def example(state): return state state = 'test' self.assertTrue( callbacks.on_forward(example).on_forward(state) == state)
def test_on_forward(self): def example(state): return state state = 'test' self.assertTrue(callbacks.on_forward(example).on_forward(state) == state)