Merge branch 'master' into blocktensor2

QuantumKitHub · Dec 11, 2024 · 4c66f24 · 4c66f24
2 parents a743813 + cd81eef
commit 4c66f24
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/src/environments/finite_envs.jl b/src/environments/finite_envs.jl
@@ -38,9 +38,10 @@ function environments(below::FiniteMPS{S}, O::DenseMPO, above=nothing) where {S}
     N = length(below)
     leftstart = isomorphism(storagetype(S),
                             left_virtualspace(below, 0) ⊗ space(O[1], 1)' ←
-                            left_virtualspace(below, 0))
+                            left_virtualspace(something(above, below), 0))
     rightstart = isomorphism(storagetype(S),
-                             right_virtualspace(below, N) ⊗ space(O[N], 4)' ←
+                             right_virtualspace(something(above, below), N) ⊗
+                             space(O[N], 4)' ←
                              right_virtualspace(below, length(below)))
     return environments(below, O, above, leftstart, rightstart)
 end
@@ -85,7 +86,8 @@ function environments(state::Union{FiniteMPS,WindowMPS}, operator::ProjectionOpe
                                         l_LL(operator.ket)[-1; -2]
     @plansor rightstart[-1; -2 -3 -4] := r_RR(operator.ket)[-1; -2] *
                                          r_RR(operator.ket)[-3; -4]
-    return environments(state, fill(nothing, length(state)), state, leftstart, rightstart)
+    return environments(state, fill(nothing, length(state)), operator.ket, leftstart,
+                        rightstart)
 end
 
 #notify the cache that we updated in-place, so it should invalidate the dependencies

diff --git a/test/algorithms.jl b/test/algorithms.jl
@@ -455,16 +455,16 @@ end
             ψ = FiniteMPS(rand, ComplexF64, len, ℙ^2, ℙ^10)
             ψ, envs, = find_groundstate(ψ, H; verbosity)
 
-            #find energy with quasiparticle ansatz
+            # find energy with quasiparticle ansatz
             energies_QP, ϕs = excitations(H, QuasiparticleAnsatz(), ψ, envs)
             @test variance(ϕs[1], H) < 1e-6
 
-            #find energy with normal dmrg
-            energies_dm, _ = excitations(H,
-                                         FiniteExcited(;
-                                                       gsalg=DMRG(; verbosity,
-                                                                  tol=1e-6, maxiter=30)), ψ)
-            @test energies_dm[1] ≈ energies_QP[1] + expectation_value(ψ, H, envs) atol = 1e-4
+            # find energy with normal dmrg
+            for gsalg in (DMRG(; verbosity, tol=1e-6),
+                          DMRG2(; verbosity, tol=1e-6, trscheme=truncbelow(1e-4)))
+                energies_dm, _ = excitations(H, FiniteExcited(; gsalg), ψ)
+                @test energies_dm[1] ≈ energies_QP[1] + expectation_value(ψ, H, envs) atol = 1e-4
+            end
 
             # find energy with Chepiga ansatz
             energies_ch, _ = excitations(H, ChepigaAnsatz(), ψ, envs)