Remove useless cast to OrderedSet

PierreQuinton · PierreQuinton · commit bb692101ba02 · 2025-03-31T18:03:33.000+02:00
diff --git a/src/torchjd/autojac/_transform/diagonalize.py b/src/torchjd/autojac/_transform/diagonalize.py
@@ -8,7 +8,7 @@
 
 class Diagonalize(Transform[Gradients, Jacobians]):
     def __init__(self, key_order: OrderedSet[Tensor]):
-        self.key_order = OrderedSet(key_order)
+        self.key_order = key_order
         self.indices: list[tuple[int, int]] = []
         begin = 0
         for tensor in self.key_order:
