Skip to content

Commit 90814db

Browse files
authored
Update VcatAtom.jl
1 parent 2f5125a commit 90814db

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/atoms/VcatAtom.jl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,10 @@ function Base.getindex(x::VcatAtom, inds::AbstractVector{<:Real})
119119
return IndexAtom(remaining, inds)
120120
end
121121

122+
function getindex(x::VcatAtom, inds::BitVector)
123+
return getindex(x, findall(inds))
124+
end
125+
122126
function Base.getindex(x::VcatAtom, inds::AbstractVector{Bool})
123-
return getindex(x, first.(filter!(last, collect(enumerate(inds)))))
127+
return getindex(x, convert(BitVector, inds))
124128
end

0 commit comments

Comments
 (0)