(self, edge_type, source, decoder, option)
| 350 | return self._undirected_edges |
| 351 | |
| 352 | def add_reverse_edges(self, edge_type, source, decoder, option): |
| 353 | self._undirected_edges.append(edge_type[2]) |
| 354 | |
| 355 | if (edge_type[0] != edge_type[1]): # pylint: disable=superfluous-parens |
| 356 | reversed_edge_type = edge_type[2] + '_reverse' |
| 357 | edge_source_reverse = self._construct_edge_source( |
| 358 | source, |
| 359 | (edge_type[1], edge_type[0], reversed_edge_type), |
| 360 | decoder, |
| 361 | direction=pywrap.Direction.REVERSED, |
| 362 | option=option) |
| 363 | self._edge_sources.append(edge_source_reverse) |
| 364 | self._edge_decoders[reversed_edge_type] = decoder |
| 365 | self._topology.add(reversed_edge_type, edge_type[1], edge_type[0]) |
| 366 | else: |
| 367 | edge_source_reverse = self._construct_edge_source( |
| 368 | source, |
| 369 | edge_type, |
| 370 | decoder, |
| 371 | direction=pywrap.Direction.REVERSED, |
| 372 | option=option) |
| 373 | self._edge_sources.append(edge_source_reverse) |
| 374 | |
| 375 | def init(self, task_index=0, task_count=1, |
| 376 | cluster="", job_name="", **kwargs): |
no test coverage detected