Fix usage of OrderedSet in _create_transform of backward

Author: PierreQuinton

tests/unit/autojac/test_backward.py

@@ -5,6 +5,7 @@
 
 from torchjd import backward
 from torchjd.aggregation import MGDA, Aggregator, Mean, Random, Sum, UPGrad
+from torchjd.autojac._transform.ordered_set import OrderedSet
 from torchjd.autojac.backward import _create_transform
 
 
@@ -20,7 +21,7 @@ def test_check_create_transform():
     transform = _create_transform(
         tensors=[y1, y2],
         aggregator=Mean(),
-        inputs={a1, a2},
+        inputs=OrderedSet([a1, a2]),
         retain_graph=False,
         parallel_chunk_size=None,
     )