(self, msg)
| 111 | self.msg_client = MsgClient(ip, port) |
| 112 | |
| 113 | def alert(self, msg): |
| 114 | if self.C.rank == 0: |
| 115 | try: |
| 116 | self.msg_client.send('[{}]: {}\n'.format(self.job_name, msg)) |
| 117 | except Exception as e: |
| 118 | print(e) |
| 119 | count = 0 |
| 120 | succ = False |
| 121 | while count < 10: |
| 122 | print('reconnecting...') |
| 123 | try: |
| 124 | if hasattr(self, 'msg_client'): |
| 125 | self.msg_client.close() |
| 126 | self.init_msg_client() |
| 127 | except Exception as e2: |
| 128 | print(e2) |
| 129 | count += 1 |
| 130 | time.sleep(1) |
| 131 | else: |
| 132 | succ = True |
| 133 | break |
| 134 | if succ: |
| 135 | self.msg_client.send('[{}]: {}'.format(self.job_name, msg)) |
| 136 | |
| 137 | def create_dataset(self): |
| 138 | ginfo = self.ginfo |
no test coverage detected