Make copy on ProjectedTTN copy-safe. Workaround for issue ITensor#98 …

…in Dictionaries.jl.
b-kloss · Jan 28, 2024 · 5a00dd6 · 5a00dd6
1 parent 05fb04c
commit 5a00dd6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/treetensornetworks/projttns/projttn.jl b/src/treetensornetworks/projttns/projttn.jl
@@ -11,7 +11,7 @@ function ProjTTN(H::TTN)
   return ProjTTN(vertices(H), H, Dictionary{edgetype(H),ITensor}())
 end
 
-copy(P::ProjTTN) = ProjTTN(P.pos, copy(P.H), copy(P.environments))
+copy(P::ProjTTN) = ProjTTN(P.pos, copy(P.H), NamedGraphs.copy_keys_values(P.environments))
 
 # trivial if we choose to specify position as above; only kept to allow using alongside
 # ProjMPO

diff --git a/src/treetensornetworks/projttns/projttn_apply.jl b/src/treetensornetworks/projttns/projttn_apply.jl
@@ -10,7 +10,7 @@ function ProjTTNApply(psi0::TTN, H::TTN)
 end
 
 function copy(P::ProjTTNApply)
-  return ProjTTNApply(P.pos, copy(P.psi0), copy(P.H), copy(P.environments))
+  return ProjTTNApply(P.pos, copy(P.psi0), copy(P.H), NamedGraphs.copy_keys_values(P.environments))
 end
 
 function set_nsite(P::ProjTTNApply, nsite)