File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 33
44from torchjd .aggregation import AlignedMTL
55
6- from ._property_testers import (
7- ExpectedStructureProperty ,
8- PermutationInvarianceProperty ,
9- StrongStationarityProperty ,
10- )
6+ from ._property_testers import ExpectedStructureProperty , PermutationInvarianceProperty
117
128
139@mark .parametrize ("aggregator" , [AlignedMTL ()])
14- class TestAlignedMTL (
15- ExpectedStructureProperty , PermutationInvarianceProperty , StrongStationarityProperty
16- ):
10+ class TestAlignedMTL (ExpectedStructureProperty , PermutationInvarianceProperty ):
1711 pass
1812
1913
Original file line number Diff line number Diff line change 88from torchjd .aggregation import CAGrad , Mean
99
1010from ._inputs import typical_matrices
11- from ._property_testers import (
12- ExpectedStructureProperty ,
13- NonConflictingProperty ,
14- StrongStationarityProperty ,
15- )
11+ from ._property_testers import ExpectedStructureProperty , NonConflictingProperty
1612
1713
1814@mark .parametrize ("aggregator" , [CAGrad (c = 0.5 )])
19- class TestCAGrad (ExpectedStructureProperty , StrongStationarityProperty ):
15+ class TestCAGrad (ExpectedStructureProperty ):
2016 pass
2117
2218
Original file line number Diff line number Diff line change 99 ExpectedStructureProperty ,
1010 NonConflictingProperty ,
1111 PermutationInvarianceProperty ,
12- StrongStationarityProperty ,
1312)
1413
1514
1615@mark .parametrize ("aggregator" , [MGDA ()])
17- class TestMGDA (
18- ExpectedStructureProperty ,
19- NonConflictingProperty ,
20- PermutationInvarianceProperty ,
21- StrongStationarityProperty ,
22- ):
16+ class TestMGDA (ExpectedStructureProperty , NonConflictingProperty , PermutationInvarianceProperty ):
2317 pass
2418
2519
You can’t perform that action at this time.
0 commit comments