b52e083827ce7aefda75c4370db0ad6942e27dae,stellar/mapper/node_mappers.py,HinSAGENodeMapper,__getitem__,#HinSAGENodeMapper#Any#,355

Before Change

        sampling_schema = self._sampling_schema
        if sampling_schema is None:
            head_node_types = set([self.schema.get_node_type(n) for n in head_nodes])
            if len(head_node_types) > 1:
                raise ValueError(
                    "Only a single head node type is currently supported for HinSAGE models"
                )
            sampling_schema = self.schema.get_sampling_layout(
                head_node_types, self.num_samples
            )

After Change

        // Resize features to (batch_size, n_neighbours, feature_size)
        batch_feats = [
            np.reshape(a, (len(head_nodes), -1 if np.size(a) > 0 else 0, a.shape[1]))
            for a in batch_feats
        ]

Instances