def setUp(self): self.data_stream = FlagDataStream(IterableDataset([1, 2, 3])) self.transformer = Transformer(self.data_stream)
def setUp(self): self.data_stream = FlagDataStream(IterableDataset([1, 2, 3])) self.transformer = Transformer( self.data_stream, produces_examples=True)