def test_text_generation_1(self):
with self.assertRaises(DLPyError):
from dlpy.applications import TextGeneration
model = TextGeneration(self.s, n_blocks=1)
model.print_summary()
def test_text_generation(self):
from dlpy.applications import TextGeneration
model = TextGeneration(self.s)
model.print_summary()
