def just_draw(self, msg, scene=[], *args, **kwargs): assert hasattr(self, 'draw'), "Model is not a drawer" episode = Episode([codraw_data.TellGroup(msg), codraw_data.ObserveCanvas(scene)]) if isinstance(self, nn.Module): self.eval() self.draw(episode, *args, **kwargs) event_multi = episode.get_last(codraw_data.DrawGroup) if event_multi is not None: return codraw_data.AbstractScene(event_multi.cliparts) event_single = episode.get_last(codraw_data.DrawClipart) return event_single.clipart
def just_tell(self, clipart, *args, **kwargs): assert hasattr(self, 'tell'), "Model is not a teller" if isinstance(self, nn.Module): self.eval() episode = Episode([codraw_data.SelectClipart(clipart)]) self.tell(episode, *args, **kwargs) return episode.get_last(codraw_data.TellGroup).msg