bug fix in partial_tr

Author: cometscome

Project.toml

@@ -1,7 +1,7 @@
 name = "LatticeMatrices"
 uuid = "dd6a91e4-736f-4540-ac85-13822ca7b545"
 authors = ["Yuki Nagai <cometscome@gmail.com>"]
-version = "0.0.3"
+version = "0.0.4"
 
 [deps]
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"

src/LinearAlgebras/linearalgebra_D.jl

@@ -3280,7 +3280,7 @@ export partial_trace
 
 
     s = zero(eltype(A))
-    if NN[μ] == position
+    if indices[μ] == position
         for ic = 1:NC
             s += A[ic, ic, indices...]
         end