Try fixing issues with diaglength

ITensor · Nov 30, 2023 · ba43de9 · ba43de9
1 parent d085d99
commit ba43de9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/NDTensors/src/dims.jl b/NDTensors/src/dims.jl
@@ -1,3 +1,5 @@
+using .DiagonalArrays: DiagonalArrays
+
 export dense, dims, dim, mindim, diaglength
 
 # dim and dims are used in the Tensor interface, overload 
@@ -26,7 +28,7 @@ mindim(inds::Tuple) = minimum(dims(inds))
 
 mindim(::Tuple{}) = 1
 
-diaglength(inds::Tuple) = mindim(inds)
+DiagonalArrays.diaglength(inds::Tuple) = mindim(inds)
 
 """
     dim_to_strides(ds)
@@ -94,4 +96,3 @@ dim(T::Tensor) = dim(inds(T))
 dim(T::Tensor, i::Int) = dim(inds(T), i)
 maxdim(T::Tensor) = maxdim(inds(T))
 mindim(T::Tensor) = mindim(inds(T))
-diaglength(T::Tensor) = mindim(T)
diff --git a/NDTensors/test/arraytensor/diagonalarray.jl b/NDTensors/test/arraytensor/diagonalarray.jl
@@ -1,5 +1,5 @@
 @eval module $(gensym())
-using NDTensors: contract
+using NDTensors: contract, tensor
 using NDTensors.SparseArrayInterface: densearray
 using NDTensors.DiagonalArrays: DiagonalArray
 using Test: @test, @testset