Merge pull request #1085 from mcabbott/one_strict

Make `OneElement` constructor safer
FluxML · Oct 2, 2021 · ade9b1e · ade9b1e · willtebbutt · Oct 2, 2021
2 parents 88f32b2 + 8acd8de
commit ade9b1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lib/array.jl b/src/lib/array.jl
@@ -50,7 +50,7 @@ struct OneElement{T,N,I,A} <: AbstractArray{T,N}
   val::T
   ind::I
   axes::A
-  OneElement(val::T, ind::I, axes::A) where {T<:Number, I<:NTuple{N,Int}, A} where {N} = new{T,N,I,A}(val, ind, axes)
+  OneElement(val::T, ind::I, axes::A) where {T<:Number, I<:NTuple{N,Int}, A<:NTuple{N,AbstractUnitRange}} where {N} = new{T,N,I,A}(val, ind, axes)
 end
 Base.size(A::OneElement) = map(length, A.axes)
 Base.axes(A::OneElement) = A.axes