From b21f6f39f41fc31efe18aa4d9d7bd818f19fcf6a Mon Sep 17 00:00:00 2001 From: Milan Date: Tue, 29 Mar 2022 12:47:44 +0100 Subject: [PATCH] use === for masked_value comparison for NaN --- src/mutual_information.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mutual_information.jl b/src/mutual_information.jl index d27cd67..afa5fa0 100644 --- a/src/mutual_information.jl +++ b/src/mutual_information.jl @@ -68,8 +68,8 @@ function bitinformation(A::AbstractArray{T}; masked_value::Union{T,Nothing}=nothing, kwargs...) where {T<:Union{Integer,AbstractFloat}} - # create a BitArray mask if a masked_value is provided - isnothing(masked_value) || return bitinformation(A,A .== masked_value;dim,kwargs...) + # create a BitArray mask if a masked_value is provided, use === to also allow NaN comparison + isnothing(masked_value) || return bitinformation(A,A .=== masked_value;dim,kwargs...) A = permute_dim_forward(A,dim) # Permute A to take adjacent entry in dimension dim n = size(A)[1] # n elements in dim