Edit set_environment for ProjTTNApply.

ITensor · Jan 30, 2024 · 9c5b42f · 9c5b42f
1 parent fefc66a
commit 9c5b42f
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/treetensornetworks/projttns/projttn_apply.jl b/src/treetensornetworks/projttns/projttn_apply.jl
@@ -33,11 +33,7 @@ function set_environments(p::ProjTTNApply, environments)
   return ProjTTNApply(pos(p), init_state(p), operator(p), environments)
 end
 
-function set_environment(p::ProjTTNApply, edge, env)
-  newenv = merge(p.environments, Dictionary((edge,), (env,)))
-  return ProjTTNApply(pos(p), init_state(p), operator(p), newenv)
-end
-
+set_environment(p::ProjTTNApply, edge, env) = set_environment!(copy(p), edge, env)
 function set_environment!(p::ProjTTNApply, edge, env)
   set!(environments(p), edge, env)
   return p