From 376e59c217ae00b9d612b754bd5a420689911659 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sergio=20S=C3=A1nchez=20Ram=C3=ADrez?=
 <sergio.sanchez.ramirez+git@bsc.es>
Date: Wed, 4 Sep 2024 18:24:53 +0200
Subject: [PATCH] Generalize `addsite!`,`rmsite!` to `AbstractQuantum`

---
 src/Quantum.jl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Quantum.jl b/src/Quantum.jl
index b6f86b576..276f65f26 100644
--- 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")