As returned object of `get_next` api of `meta_path_sampler`.
| 681 | |
| 682 | |
| 683 | class Layers(object): |
| 684 | """ As returned object of `get_next` api of `meta_path_sampler`. |
| 685 | """ |
| 686 | |
| 687 | def __init__(self, layers=None): |
| 688 | self.layers = layers if layers else [] |
| 689 | |
| 690 | def layer(self, layer_id): |
| 691 | """ Get one `Layer`. |
| 692 | """ |
| 693 | layer_id -= 1 |
| 694 | if isinstance(self.layers, list) and layer_id < len(self.layers): |
| 695 | return self.layers[layer_id] |
| 696 | else: |
| 697 | raise ValueError("layer id beyond the layers length.") |
| 698 | |
| 699 | def layer_size(self, layer_id): |
| 700 | """ Get size of the given `Layer`. |
| 701 | """ |
| 702 | layer_id -= 1 |
| 703 | if isinstance(self.layers, list) and layer_id < len(self.layers): |
| 704 | return self.layers[layer_id].shape |
| 705 | else: |
| 706 | raise ValueError("layer id beyond the layers length.") |
| 707 | |
| 708 | def layer_nodes(self, layer_id): |
| 709 | """ Get `Nodes` of the given `Layer`. |
| 710 | """ |
| 711 | layer_id -= 1 |
| 712 | if isinstance(self.layers, list) and layer_id < len(self.layers): |
| 713 | return self.layers[layer_id].nodes |
| 714 | else: |
| 715 | raise ValueError("layer id beyond the layers length.") |
| 716 | |
| 717 | def layer_edges(self, layer_id): |
| 718 | """ Get `Edges` of the given `Layer`. |
| 719 | """ |
| 720 | layer_id -= 1 |
| 721 | if isinstance(self.layers, list) and layer_id < len(self.layers): |
| 722 | return self.layers[layer_id].edges |
| 723 | else: |
| 724 | raise ValueError("layer id beyond the layers length.") |
| 725 | |
| 726 | def set_layer_nodes(self, layer_id, nodes): |
| 727 | """ Set `Nodes` of the given `Layer`. |
| 728 | """ |
| 729 | layer_id -= 1 |
| 730 | if isinstance(self.layers, list) and layer_id < len(self.layers): |
| 731 | if isinstance(self.layers[layer_id], Layer): |
| 732 | self.layers[layer_id].set_nodes(nodes) |
| 733 | else: |
| 734 | raise ValueError("layer {} is not a SingleLayer".format(layer_id)) |
| 735 | else: |
| 736 | raise ValueError("layer id beyond the layers length.") |
| 737 | |
| 738 | def set_layer_edges(self, layer_id, edges): |
| 739 | """ Set `Edges` of the given `Layer`. |
| 740 | """ |