Try fixing tests

ITensor · Nov 14, 2024 · 1f736ef · 1f736ef
1 parent cae9bfc
commit 1f736ef
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/itensor.jl b/src/itensor.jl
@@ -1,4 +1,4 @@
-using SparseArrays: SparseArrays, nnz
+using NDTensors: NDTensors, nnz
 using .TagSets: TagSets, hastags, replacetags
 
 # Private inner constructor
@@ -853,7 +853,11 @@ hasqns(T::Union{Tensor,ITensor}) = hasqns(inds(T))
 
 eachnzblock(T::ITensor) = eachnzblock(tensor(T))
 
-SparseArrays.nnz(T::ITensor) = nnz(tensor(T))
+# TODO: Switch this to `SparseArrays.nnz`, it is written
+# this way for backwards compatibility since older versions
+# of NDTensors had their own `NDTensors.nnz` function
+# that didn't overload `SparseArrays.nnz`.
+NDTensors.nnz(T::ITensor) = nnz(tensor(T))
 
 nblocks(T::ITensor, args...) = nblocks(tensor(T), args...)