b50ca39a02936ba01324abee3adfd84a3ffdd057,demos/genes/data.py,GeneGraph,get_batch,#GeneGraph#Any#Any#,94

Before Change

        flatten = lambda l: [item for sublist in l for item in sublist]
        coex, ppi, epis = self.sample_neighs(indices, ns[0])
        coex_1 = self.sample_neighs(flatten(coex), ns[1])
        ppi_1 = self.sample_neighs(flatten(ppi), ns[1])
        epis_1 = self.sample_neighs(flatten(epis), ns[1])
        return (
            self.get_labels(indices),

After Change

        // YT: somewhat generalized version of the commented-out snippet above:
        neigh_1hop = self.sample_neighs(indices, ns[0])
        neigh_2hop = dict()
        for i, et in enumerate(self.adj.keys()):
            neigh_2hop[et] = self.sample_neighs(flatten(neigh_1hop[i]), ns[1])

        return (
            self.get_labels(indices),
            [
                self.get_feats(flatten(inds)).reshape([nb, -1, self.feats.shape[1]])

Instances