hiro/two_qubit_model
1
0
Fork 0
mirror of https://github.com/vale981/two_qubit_model synced 2025-03-06 02:01:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

support reading the multi bath model from the db

Browse source
This commit is contained in:
Valentin Boettcher 2022-08-26 09:51:37 +02:00
parent 1475df6dd6
commit 23aea98c45
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
hiro_models/model_auxiliary.py
View file

@ -11,7 +11,7 @@ from contextlib import contextmanager
from .utility import JSONEncoder, object_hook from .utility import JSONEncoder, object_hook
from filelock import FileLock from filelock import FileLock
from pathlib import Path from pathlib import Path
from .one_qubit_model import QubitModel from .one_qubit_model import QubitModel, QubitModelMutliBath
from .two_qubit_model import TwoQubitModel from .two_qubit_model import TwoQubitModel
from collections.abc import Sequence, Iterator, Iterable from collections.abc import Sequence, Iterator, Iterable
import shutil import shutil
@ -70,6 +70,9 @@ def model_hook(dct: dict[str, Any]):
if model == "TwoQubitModel": if model == "TwoQubitModel":
return TwoQubitModel.from_dict(treated_vals) return TwoQubitModel.from_dict(treated_vals)
if model == "QubitModelMutliBath":
return QubitModelMutliBath.from_dict(treated_vals)
return object_hook(dct) return object_hook(dct)