Skip to content

Commit ca21dd9

Browse files
authored
LazyStack.lazystack
1 parent e987965 commit ca21dd9

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/stack.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11

2-
import LazyStack
2+
using LazyStack
33

44
# for stack_iter
55
LazyStack.no_wraps(a::KeyedArray) = LazyStack.no_wraps(NamedDims.unname(parent(a)))
@@ -14,35 +14,35 @@ function LazyStack.rewrap_like(A, a::NamedTuple)
1414
end
1515

1616
# tuple of arrays
17-
function LazyStack.stack(x::Tuple{Vararg{<:KeyedArray}})
17+
function LazyStack.lazystack(x::Tuple{Vararg{<:KeyedArray}})
1818
KeyedArray(LazyStack.stack(map(parent, x)), stack_keys(x))
1919
end
2020

2121
stack_keys(xs::Tuple{Vararg{<:KeyedArray}}) =
2222
(keys_or_axes(first(xs))..., Base.OneTo(length(xs)))
2323

2424
# array of arrays: first strip off outer containers...
25-
function LazyStack.stack(xs::KeyedArray{<:AbstractArray})
25+
function LazyStack.lazystack(xs::KeyedArray{<:AbstractArray})
2626
KeyedArray(LazyStack.stack(parent(xs)), stack_keys(xs))
2727
end
28-
function LazyStack.stack(xs::KeyedArray{<:AbstractArray,N,<:NamedDimsArray{L}}) where {L,N}
28+
function LazyStack.lazystack(xs::KeyedArray{<:AbstractArray,N,<:NamedDimsArray{L}}) where {L,N}
2929
data = LazyStack.stack(parent(parent(xs)))
3030
KeyedArray(LazyStack.ensure_named(data, LazyStack.getnames(xs)), stack_keys(xs))
3131
end
32-
function LazyStack.stack(xs::NamedDimsArray{L,<:AbstractArray,N,<:KeyedArray}) where {L,N}
32+
function LazyStack.lazystack(xs::NamedDimsArray{L,<:AbstractArray,N,<:KeyedArray}) where {L,N}
3333
data = LazyStack.stack(parent(parent(xs)))
3434
LazyStack.ensure_named(KeyedArray(data, stack_keys(xs)), LazyStack.getnames(xs))
3535
end
3636

3737
# ... then deal with inner ones:
38-
function LazyStack.stack(x::AT) where {AT <: AbstractArray{<:KeyedArray{T,IN},ON}} where {T,IN,ON}
38+
function LazyStack.lazystack(x::AT) where {AT <: AbstractArray{<:KeyedArray{T,IN},ON}} where {T,IN,ON}
3939
KeyedArray(LazyStack.Stacked{T, IN+ON, AT}(x), stack_keys(x))
4040
end
41-
function LazyStack.stack(x::AT) where {AT <: AbstractArray{<:KeyedArray{T,IN,IT},ON}} where {IT<:NamedDimsArray} where {T,IN,ON}
41+
function LazyStack.lazystack(x::AT) where {AT <: AbstractArray{<:KeyedArray{T,IN,IT},ON}} where {IT<:NamedDimsArray} where {T,IN,ON}
4242
data = LazyStack.Stacked{T, IN+ON, AT}(x)
4343
KeyedArray(NamedDimsArray(data, LazyStack.getnames(x)), stack_keys(x))
4444
end
45-
function LazyStack.stack(x::AT) where {AT <: AbstractArray{<:NamedDimsArray{L,T,IN,IT},ON}} where {IT<:KeyedArray} where {T,IN,ON,L}
45+
function LazyStack.lazystack(x::AT) where {AT <: AbstractArray{<:NamedDimsArray{L,T,IN,IT},ON}} where {IT<:KeyedArray} where {T,IN,ON,L}
4646
data = LazyStack.Stacked{T, IN+ON, AT}(x)
4747
NamedDimsArray(KeyedArray(data, stack_keys(x)), LazyStack.getnames(x))
4848
end

0 commit comments

Comments
 (0)