Update test/nn/aggr/test_scaler.py

pyg-team · Feb 6, 2023 · 9bf0fbd · 9bf0fbd
1 parent 0b97e00
commit 9bf0fbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/nn/aggr/test_scaler.py b/test/nn/aggr/test_scaler.py
@@ -4,7 +4,7 @@
 from torch_geometric.nn import DegreeScalerAggregation
 
 
-@pytest.mark.parametrize("train_norm", [True, False])
+@pytest.mark.parametrize('train_norm', [True, False])
 def test_degree_scaler_aggregation(train_norm):
     x = torch.randn(6, 16)
     index = torch.tensor([0, 0, 1, 1, 1, 2])