cf38c49c6260e2a7af3c1f752088532f0d76ab6d,torch_geometric/nn/conv/message_passing.py,MessagePassing,__init__,#MessagePassing#Any#Any#Any#,53

Before Change

        self.__message_signature__ = inspect.signature(self.message)
        // skip self, out
        self.__update_signature__ = inspect.signature(self.update)
        if set(update_special_args) - set(
            self.__update_signature__.parameters
        ):
            raise TypeError(
                "Incomplete signature of update: {} are "
                "missing required arguments".format(
                    (
                        set(update_special_args)
                        - set(self.__update_signature__.parameters)
                    )
                )
            )
        self.__aggregate_signature__ = inspect.signature(self.aggregate)
        if set(aggregate_special_args) - set(
            self.__aggregate_signature__.parameters
        ):
            raise TypeError(
                "Incomplete signature of aggregate: {} are "
                "missing required arguments".format(
                    set(aggregate_special_args)
                    - set(self.__aggregate_signature__.parameters)
                )
            )

After Change

        aggr_args = set(self.__aggr_sign__.keys()) - aggr_special_args
        update_args = set(self.__update_sign__.keys()) - update_special_args

        self.__args__ = set().union(msg_args, aggr_args, update_args)

    def __set_size__(self, size, index, tensor):
        if size[index] is None:
            size[index] = tensor.size(self.node_dim)

Instances