From 629af711e5931b36d85d965777b372fa141c60b9 Mon Sep 17 00:00:00 2001
From: Joey <joeytindall94@googlemail.com>
Date: Fri, 8 Nov 2024 10:02:21 -0500
Subject: [PATCH] Wrap tree orthogonalize for ttns

---
 src/apply.jl                                        | 4 ++--
 src/tebd.jl                                         | 2 +-
 src/treetensornetworks/abstracttreetensornetwork.jl | 4 ++++
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/apply.jl b/src/apply.jl
index d38f04f9..6a55f45f 100644
--- a/src/apply.jl
+++ b/src/apply.jl
@@ -200,7 +200,7 @@ function ITensors.apply(
   v⃗ = neighbor_vertices(ψ, o)
   if length(v⃗) == 1
     if ortho
-      ψ = orthogonalize(ψ, v⃗[1])
+      ψ = tree_orthogonalize(ψ, v⃗[1])
     end
     oψᵥ = apply(o, ψ[v⃗[1]])
     if normalize
@@ -215,7 +215,7 @@ function ITensors.apply(
       error("Vertices where the gates are being applied must be neighbors for now.")
     end
     if ortho
-      ψ = orthogonalize(ψ, v⃗[1])
+      ψ = tree_orthogonalize(ψ, v⃗[1])
     end
     if variational_optimization_only || !is_product_env
       ψᵥ₁, ψᵥ₂ = full_update_bp(
diff --git a/src/tebd.jl b/src/tebd.jl
index edf5a188..d1d96017 100644
--- a/src/tebd.jl
+++ b/src/tebd.jl
@@ -23,7 +23,7 @@ function tebd(
     ψ = apply(u⃗, ψ; cutoff, maxdim, normalize=true, ortho, kwargs...)
     if ortho
       for v in vertices(ψ)
-        ψ = orthogonalize(ψ, v)
+        ψ = tree_orthogonalize(ψ, v)
       end
     end
   end
diff --git a/src/treetensornetworks/abstracttreetensornetwork.jl b/src/treetensornetworks/abstracttreetensornetwork.jl
index a08fe27d..4635ac8a 100644
--- a/src/treetensornetworks/abstracttreetensornetwork.jl
+++ b/src/treetensornetworks/abstracttreetensornetwork.jl
@@ -50,6 +50,10 @@ function ITensorMPS.orthogonalize(ttn::AbstractTTN, region; kwargs...)
   return orthogonalize(ttn, [region]; kwargs...)
 end
 
+function tree_orthogonalize(ttn::AbstractTTN, args...; kwargs...)
+  return orthogonalize(ttn, args...; kwargs...)
+end
+
 # 
 # Truncation
 #