From 4a2362e0c16b4582a46b615136dd681fe99f96dd Mon Sep 17 00:00:00 2001
From: MatteoRobbiati <robbiati.matteo@gmail.com>
Date: Mon, 3 Feb 2025 16:57:50 +0400
Subject: [PATCH] feat: add set_device method to TN abstract backend

---
 src/qibotn/backends/abstract.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/qibotn/backends/abstract.py b/src/qibotn/backends/abstract.py
index 1ee3c5b..9b77f20 100644
--- a/src/qibotn/backends/abstract.py
+++ b/src/qibotn/backends/abstract.py
@@ -30,6 +30,9 @@ class QibotnBackend(NumpyBackend):
         if precision != self.precision:
             super().set_precision(precision)
 
+    def set_device(self, device):
+        self.device = device
+
     # @abstractmethod
     def configure_tn_simulation(self, **config):
         """Configure the TN simulation that will be performed."""