(self, output)
| 105 | super().__exit__(*args) |
| 106 | |
| 107 | def add_stop_condition(self, output): |
| 108 | if isinstance(output, core.BlobReference): |
| 109 | t = Task(outputs=[output], group=self.epoch_group) |
| 110 | output = t.outputs()[0] |
| 111 | assert isinstance(output, TaskOutput) |
| 112 | self.stop_conditions.append(output) |
| 113 | |
| 114 | |
| 115 | def get_ckpt_filename(node_name, epoch): |