alter JordanMPOTensor Adapt implementation

Author: lkdvos

ext/MPSKitAdaptExt.jl

@@ -32,26 +32,9 @@ end
 
 Adapt.adapt_structure(to, mpo::MPO) = MPO(map(adapt(to), mpo.O))
 
-function Adapt.adapt_structure(::Type{TorA}, W::MPSKit.JordanMPOTensor) where {TorA <: Union{Number, DenseVector{<:Number}}}
-    TT = MPSKit.jordanmpotensortype(spacetype(W), TorA)
-    W′ = TT(undef, space(W))
-    ad = adapt(TorA)
-
-    for (k, v) in nonzero_pairs(W.A)
-        W′.A[k] = ad(v)
-    end
-    for (k, v) in nonzero_pairs(W.B)
-        W′.B[k] = ad(v)
-    end
-    for (k, v) in nonzero_pairs(W.C)
-        W′.C[k] = ad(v)
-    end
-    for (k, v) in nonzero_pairs(W.D)
-        W′.D[k] = ad(v)
-    end
-
-    return W′
-end
+Adapt.adapt_structure(to, W::MPSKit.JordanMPOTensor) =
+    MPSKit.JordanMPOTensor(space(W), adapt(to, W.A), adapt(to, W.B), adapt(to, W.C), adapt(to, W.D))
+
 Adapt.adapt_structure(to, mpo::MPOHamiltonian) = MPOHamiltonian(map(adapt(to), mpo.W))
 
 end