Checking self.normalize once

pyg-team · Dec 11, 2024 · d22db1d · d22db1d
1 parent 8bbdbc4
commit d22db1d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/torch_geometric/nn/conv/gat_conv.py b/torch_geometric/nn/conv/gat_conv.py
@@ -509,7 +509,6 @@ def forward(  # noqa: F811
                         "The usage of 'normalize' is not supported "
                         "for bipartite message passing.")
 
-        if self.normalize:
             if isinstance(edge_index, Tensor):
                 edge_index, edge_attr = remove_self_loops(
                     edge_index, edge_attr)

diff --git a/torch_geometric/nn/conv/gatv2_conv.py b/torch_geometric/nn/conv/gatv2_conv.py
@@ -338,7 +338,6 @@ def forward(  # noqa: F811
                         "The usage of 'normalize' is not supported "
                         "for bipartite message passing.")
 
-        if self.normalize:
             if isinstance(edge_index, Tensor):
                 edge_index, edge_attr = remove_self_loops(
                     edge_index, edge_attr)