Update infarrays.jl

JuliaArrays · Mar 7, 2021 · 2dddc9d · 2dddc9d
1 parent 4c99bf2
commit 2dddc9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/infarrays.jl b/src/infarrays.jl
@@ -234,8 +234,8 @@ end
 # Diagonal
 #####
 
-one(D::Diagonal{T,<:AbstractFill{T,Tuple{OneToInf{Int}}}) where T = Eye{T}(size(D,1))
-copy(D::Diagonal{T,<:AbstractFill{T,Tuple{OneToInf{Int}}}) where T = D
+one(D::Diagonal{T,<:AbstractFill{T,1,Tuple{OneToInf{Int}}}}) where T = Eye{T}(size(D,1))
+copy(D::Diagonal{T,<:AbstractFill{T,1,Tuple{OneToInf{Int}}}}) where T = D
 
 BroadcastStyle(::Type{<:Diagonal{<:Any,<:AbstractInfUnitRange}}) = LazyArrayStyle{2}()