Ensure inference of Bool when array type is Any (#45452)

JuliaLang · Jun 1, 2022 · ab4c8cf · ab4c8cf
1 parent cbcb359
commit ab4c8cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/base/array.jl b/base/array.jl
@@ -2546,7 +2546,7 @@ function filter(f, a::Array{T, N}) where {T, N}
  b = Vector{T}(undef, length(a))
  for ai in a
  @inbounds b[j] = ai
- j = ifelse(f(ai), j+1, j)
+ j = ifelse(f(ai)::Bool, j+1, j)
  end
  resize!(b, j-1)
  sizehint!(b, length(b))
@@ -2561,7 +2561,7 @@ function filter(f, a::AbstractArray)
  for idx in eachindex(a)
  @inbounds idxs[j] = idx
  ai = @inbounds a[idx]
- j = ifelse(f(ai), j+1, j)
+ j = ifelse(f(ai)::Bool, j+1, j)
  end
  resize!(idxs, j-1)
  res = a[idxs]
@@ -2591,7 +2591,7 @@ function filter!(f, a::AbstractVector)
  j = firstindex(a)
  for ai in a
  @inbounds a[j] = ai
- j = ifelse(f(ai), nextind(a, j), j)
+ j = ifelse(f(ai)::Bool, nextind(a, j), j)
  end
  j > lastindex(a) && return a
  if a isa Vector