Skip to content

Commit 0ba9081

Browse files
committed
chore: refactor Device to contain StoredBackendInfo
1 parent fd76b02 commit 0ba9081

File tree

3 files changed

+6
-7
lines changed

3 files changed

+6
-7
lines changed

integration/test_devices.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""Test basic functionality relating to the device module."""
22

33
from pytket.backends.backendinfo import BackendInfo
4+
from quantinuum_schemas.models.backend_info import StoredBackendInfo
45

56
import qnexus as qnx
67
from qnexus.models import Device
@@ -16,7 +17,7 @@ def test_device_get_all(
1617

1718
for device in devices:
1819
assert isinstance(device, Device)
19-
assert device.backend_info_dict is not None # pylint: disable=protected-access
20+
assert isinstance(device.stored_backend_info, StoredBackendInfo)
2021
assert isinstance(device.backend_info, BackendInfo)
2122

2223

qnexus/client/devices.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,7 @@ def get_all( # pylint: disable=too-many-positional-arguments
7373
backend_name=backend_name,
7474
device_name=backend_info["device_name"],
7575
nexus_hosted=backendinfolist["is_local"],
76-
backend_info_dict=StoredBackendInfo(**backend_info)
77-
.to_pytket_backend_info()
78-
.to_dict(),
76+
stored_backend_info=StoredBackendInfo(**backend_info),
7977
)
8078
)
8179

qnexus/models/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,12 @@ class Device(BaseModel):
8181
backend_name: str
8282
device_name: Optional[str]
8383
nexus_hosted: bool
84-
backend_info_dict: dict[str, Any]
84+
stored_backend_info: StoredBackendInfo
8585

8686
@property
8787
def backend_info(self) -> BackendInfo:
8888
"""The BackendInfo for the Device."""
89-
return BackendInfo.from_dict(self.backend_info_dict)
89+
return self.stored_backend_info.to_pytket_backend_info()
9090

9191
def df(self) -> pd.DataFrame:
9292
"""Present in a pandas DataFrame."""
@@ -95,7 +95,7 @@ def df(self) -> pd.DataFrame:
9595
"backend_name": self.backend_name,
9696
"device_name": self.device_name,
9797
"nexus_hosted": self.nexus_hosted,
98-
"backend_info": self.backend_info_dict,
98+
"backend_info": self.stored_backend_info.to_pytket_backend_info(),
9999
},
100100
index=[0],
101101
)

0 commit comments

Comments
 (0)