refactor(autojac): Remove redundant cast in _AggregateMatrices (#295)

* Remove redundant cast of key_order to OrderedSet in the constructor of _AggregateMatrices

Author: ValerianRey

src/torchjd/autojac/_transform/aggregate.py

@@ -32,7 +32,7 @@ def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
 
 class _AggregateMatrices(Transform[JacobianMatrices, GradientVectors]):
     def __init__(self, aggregator: Aggregator, key_order: OrderedSet[Tensor]):
-        self.key_order = OrderedSet(key_order)
+        self.key_order = key_order
         self.aggregator = aggregator
 
     def __call__(self, jacobian_matrices: JacobianMatrices) -> GradientVectors: