fixup

Author: oscarvanderheide

src/operators/epg.jl

@@ -289,7 +289,7 @@ size/type as Ω s.t. later on the decay can be applied by element-wise multiplic
 - `Ω`: The configuration state matrix
 - `D`: The diffusion coefficient
 """
-@inline function diffusion_decay_matrix(Ω::AbstractConfigurationStates{Complex{T}}, D::T) where {T}
+@inline function diffusion_decay_matrix(Ω::AbstractConfigurationStates, D::Real)
     expbD = similar(Ω)
     for state in 0:size(Ω, 2)-1
         bᵀD = T(((state + 0.5)^2 + 1.0 / 12.0) * D)
@@ -308,7 +308,7 @@ On GPU, when each thread only holds parts of the configuration state matrix, eac
 computes the corresponding columns of the diffusion decay matrix. This is the reason why 
 the `ConfigurationStatesSubset` has a field `orders`.
 """
-@inline function diffusion_decay_matrix(Ω::ConfigurationStatesSubset{Complex{T}}, D::T) where {T}
+@inline function diffusion_decay_matrix(Ω::ConfigurationStatesSubset, D::Real)
     expbD = similar(Ω)
     idx = 0
     for state in Ω.orders
@@ -331,7 +331,7 @@ the pre-computed diffusion decay matrix.
 - `Ω`: The configuration state matrix
 - `diffusion_decay_matrix`: The pre-calculated diffusion decay terms
 """
-@inline function diffuse!(Ω::AbstractConfigurationStates{Complex{T}}, diffusion_decay_matrix::T) where {T}
+@inline function diffuse!(Ω::AbstractConfigurationStates, diffusion_decay_matrix::AbstractConfigurationStates)
     Ω .*= diffusion_decay_matrix
     return nothing
 end