Flatten `BatchGraph` to numpy array in the order of [edge_index, nodes, graph_node_offsets, edges, graph_edge_offsets, additional_keys]
(self)
| 158 | pass |
| 159 | |
| 160 | def flatten(self): |
| 161 | """ Flatten `BatchGraph` to numpy array in the order of |
| 162 | [edge_index, nodes, graph_node_offsets, edges, graph_edge_offsets, |
| 163 | additional_keys] |
| 164 | """ |
| 165 | def append_flatten_data(flatten_list, data) : |
| 166 | data_list = [data.int_attrs, data.float_attrs, data.string_attrs, |
| 167 | data.labels, data.weights, data.ids] |
| 168 | flatten_list.extend([item for item in data_list if item is not None]) |
| 169 | return flatten_list |
| 170 | |
| 171 | flatten_list = [self.edge_index] |
| 172 | flatten_list = append_flatten_data(flatten_list, self.nodes) |
| 173 | flatten_list.append(self.graph_node_offsets) |
| 174 | if self.edges is not None: |
| 175 | flatten_list = append_flatten_data(flatten_list, self.edges) |
| 176 | flatten_list.append(self.graph_edge_offsets) |
| 177 | for key in self.additional_keys: |
| 178 | flatten_list.append(self[key]) |
| 179 | return flatten_list |
| 180 | |
| 181 | @classmethod |
| 182 | def from_graphs(cls, graphs, additional_keys=[]): |
no test coverage detected