improce code style

euro-hpc-pl · Nov 20, 2020 · a3e631a · a3e631a
1 parent b7c463d
commit a3e631a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/search.jl b/src/search.jl
@@ -22,7 +22,7 @@ function spectrum(ψ::MPS, keep::Int)
     pCut = prob = 0.
     k = 1
 
-    if keep < (*)(rank(ψ)...)
+    if keep < prod(rank(ψ))
         keep_extra += 1
     end
 

diff --git a/src/utils.jl b/src/utils.jl
@@ -9,8 +9,8 @@ _σ(idx::Int) = (idx == 1) ? -1 : idx - 1
 
 local_basis(d::Int) = union(-1, 1:d-1)
 
-function proj(state::T, dims::S) where {T, S <: Union{Vector, NTuple}}
-    P = [] 
+function proj(state, dims::T) where {T <: Union{Vector, NTuple}}
+    P = Matrix{Float64}[] 
     for (σ, r) ∈ zip(state, dims)
         v = zeros(r)
         v[idx(σ)...] = 1.