Fix type signatures of entropy functions

dev-ket · Jun 27, 2024 · 96b290d · 96b290d
1 parent 1203ee6
commit 96b290d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/entropy.jl b/src/entropy.jl
@@ -1,4 +1,4 @@
-_log(base::Real, x) = x > 0 ? log(base, x) : zero(x)
+_log(base::Real, x::Real) = x > 0 ? log(base, x) : zero(x)
 
 """
     relative_entropy([base=2,] ρ::AbstractMatrix, σ::AbstractMatrix)
@@ -7,7 +7,7 @@ Computes the (quantum) relative entropy tr(`ρ` (log `ρ` - log `σ`)) between p
 
 Reference: [Quantum relative entropy](https://en.wikipedia.org/wiki/Quantum_relative_entropy).
 """
-function relative_entropy(base::Real, ρ::AbstractMatrix{T}, σ::AbstractMatrix{S}) where {T,S}
+function relative_entropy(base::Real, ρ::AbstractMatrix{T}, σ::AbstractMatrix{S}) where {T<:Real,S<:Real}
     R = real(promote_type(T, S))
     if size(ρ) != size(σ)
         throw(ArgumentError("ρ and σ have the same size."))
@@ -40,7 +40,7 @@ Computes the relative entropy D(`p`||`q`) = Σᵢpᵢlog(pᵢ/qᵢ) between two
 
 Reference: [Relative entropy](https://en.wikipedia.org/wiki/Relative_entropy).
 """
-function relative_entropy(base::Real, p::AbstractVector{T}, q::AbstractVector{S}) where {T,S<:Real}
+function relative_entropy(base::Real, p::AbstractVector{T}, q::AbstractVector{S}) where {T<Real,S<:Real}
     R = promote_type(T, S)
     if length(p) != length(q)
         throw(ArgumentError("`p` and q must have the same length."))