1dbc4bf21b2f67189f3a0c761a774ccaadb47587,stellar/mapper/node_mappers.py,GraphSAGENodeMapper,__init__,#GraphSAGENodeMapper#Any#Any#Any#Any#Any#Any#Any#,49

Before Change

            ["feature" in vdata for v, vdata in G.nodes(data=True)]
        )
        if not nodes_have_features:
            print("Warning: Not all nodes have a "feature" attribute.")
            print("Warning: This is required for the GraphSAGE mapper.")

        // Check that all nodes have features of the same size
        // Note: if there are no features in the nodes this will be 1!

After Change

        }

        // Check all nodes have features
        if None in feature_sizes:
            raise RuntimeError(
                "Not all nodes have a "feature" attribute: "
                "this is required for the GraphSAGE mapper."
            )

        // Sanity checks on feature sizes
        if feature_size:
            if feature_size not in feature_sizes:
                raise RuntimeWarning(
                    "Specified feature size doesn"t match graph features"

Instances