def setUp(self): super(TestRNNDecoderWithRNNCellGRUClass, self).setUp() self.model = self.create_model(GRU)