Generalize addsite!,rmsite! to AbstractQuantum

bsc-quantic · Sep 4, 2024 · 376e59c · 376e59c
1 parent 5d3ff97
commit 376e59c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/Quantum.jl b/src/Quantum.jl
@@ -209,6 +209,7 @@ function LinearAlgebra.adjoint!(tn::AbstractQuantum)
     return tn
 end
 
+addsite!(tn::AbstractQuantum, site, index) = addsite!(Quantum(tn), site, index)
 function addsite!(tn::Quantum, site, index)
     if haskey(tn.sites, site)
         error("Site $site already exists")
@@ -221,6 +222,7 @@ function addsite!(tn::Quantum, site, index)
     return tn.sites[site] = index
 end
 
+rmsite!(tn::AbstractQuantum, site) = rmsite!(Quantum(tn), site)
 function rmsite!(tn::Quantum, site)
     if !haskey(tn.sites, site)
         error("Site $site does not exist")