| 788 | self._shape = shape |
| 789 | |
| 790 | class SubGraph(object): |
| 791 | def __init__(self, edge_index, nodes, edges=None, **kwargs): |
| 792 | self._nodes = nodes |
| 793 | self._edge_index = edge_index |
| 794 | self._edges = edges |
| 795 | for key, item in kwargs.items(): |
| 796 | self[key] = item |
| 797 | |
| 798 | @property |
| 799 | def nodes(self): |
| 800 | return self._nodes |
| 801 | |
| 802 | @property |
| 803 | def edge_index(self): |
| 804 | return self._edge_index |
| 805 | |
| 806 | @property |
| 807 | def edges(self): |
| 808 | return self._edges |
| 809 | |
| 810 | def __getitem__(self, key): |
| 811 | return getattr(self, key, None) |
| 812 | |
| 813 | def __setitem__(self, key, value): |
| 814 | setattr(self, key, value) |