(self, g)
| 1796 | return g |
| 1797 | |
| 1798 | def raw(self, g): |
| 1799 | use_eweight = False |
| 1800 | if (self.eweight_name is not None) and self.eweight_name in g.edata: |
| 1801 | use_eweight = True |
| 1802 | |
| 1803 | feat_list = [] |
| 1804 | with g.local_scope(): |
| 1805 | if use_eweight: |
| 1806 | message_func = fn.u_mul_e( |
| 1807 | self.in_feat_name, self.eweight_name, "m" |
| 1808 | ) |
| 1809 | else: |
| 1810 | message_func = fn.copy_u(self.in_feat_name, "m") |
| 1811 | for _ in range(self.k): |
| 1812 | g.update_all(message_func, fn.sum("m", self.in_feat_name)) |
| 1813 | feat_list.append(g.ndata[self.in_feat_name]) |
| 1814 | return feat_list |
| 1815 | |
| 1816 | def rw(self, g): |
| 1817 | use_eweight = False |
nothing calls this directly
no test coverage detected