def METHOD_NAME(self, x_j: Optional[Tensor], pos_i: Tensor,
            pos_j: Tensor) -> Tensor:
    msg = pos_j - pos_i
    if x_j is not None:
        msg = torch.cat([x_j, msg], dim=1)
    if self.local_nn is not None:
        msg = self.local_nn(msg)
    return msg