update

pyg-team · Jun 18, 2024 · 84a6da9 · 84a6da9
1 parent 8380b5f
commit 84a6da9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/nn/aggr/test_attention.py b/test/nn/aggr/test_attention.py
@@ -4,8 +4,10 @@
 import torch_geometric.typing
 from torch_geometric.nn import MLP
 from torch_geometric.nn.aggr import AttentionalAggregation
+from torch_geometric.testing import withPackage
 
 
+@withPackage('torch-scatter')
 @pytest.mark.parametrize('dim', [2, 3])
 def test_attentional_aggregation(dim):
     channels = 16