def finalize_episode(self):
    print('\nCHAT DONE.\n')
    if self.display_partner_persona:
        partner_persona = self.p2.replace('your persona:', "partner's persona:")
        print(f'''Your partner was playing the following persona:
{partner_persona}''')
    if (not self.epoch_done()):
        print('[ Preparing new chat ... ]\n')
def finalize_episode(self):
    print('\nCHAT DONE.\n')
    if self.display_partner_persona:
        partner_persona = self.p2.replace('your persona:', "partner's persona:")
        print(f'Your partner was playing the following persona:
{partner_persona}')
    if (not self.epoch_done()):
        print('[ Preparing new chat ... ]\n')