Add failing SSProperty to AMTL CAGRAD and MGDA

Author: ValerianRey

tests/unit/aggregation/test_aligned_mtl.py

@@ -3,11 +3,17 @@
 
 from torchjd.aggregation import AlignedMTL
 
-from ._property_testers import ExpectedStructureProperty, PermutationInvarianceProperty
+from ._property_testers import (
+    ExpectedStructureProperty,
+    PermutationInvarianceProperty,
+    StrongStationarityProperty,
+)
 
 
 @mark.parametrize("aggregator", [AlignedMTL()])
-class TestAlignedMTL(ExpectedStructureProperty, PermutationInvarianceProperty):
+class TestAlignedMTL(
+    ExpectedStructureProperty, PermutationInvarianceProperty, StrongStationarityProperty
+):
     pass
 
 

tests/unit/aggregation/test_cagrad.py

@@ -8,11 +8,15 @@
 from torchjd.aggregation import CAGrad, Mean
 
 from ._inputs import typical_matrices
-from ._property_testers import ExpectedStructureProperty, NonConflictingProperty
+from ._property_testers import (
+    ExpectedStructureProperty,
+    NonConflictingProperty,
+    StrongStationarityProperty,
+)
 
 
 @mark.parametrize("aggregator", [CAGrad(c=0.5)])
-class TestCAGrad(ExpectedStructureProperty):
+class TestCAGrad(ExpectedStructureProperty, StrongStationarityProperty):
     pass
 
 

tests/unit/aggregation/test_mgda.py

@@ -9,11 +9,17 @@
     ExpectedStructureProperty,
     NonConflictingProperty,
     PermutationInvarianceProperty,
+    StrongStationarityProperty,
 )
 
 
 @mark.parametrize("aggregator", [MGDA()])
-class TestMGDA(ExpectedStructureProperty, NonConflictingProperty, PermutationInvarianceProperty):
+class TestMGDA(
+    ExpectedStructureProperty,
+    NonConflictingProperty,
+    PermutationInvarianceProperty,
+    StrongStationarityProperty,
+):
     pass