hiro/two_qubit_model
1
0
Fork 0
mirror of https://github.com/vale981/two_qubit_model synced 2025-03-04 17:21:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

add patch functionality to the db migration

Browse source
This commit is contained in:
Valentin Boettcher 2023-07-07 16:05:40 -04:00
parent 12caba7a64
commit fc67e60972
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
hiro_models/model_auxiliary.py
View file

@ -14,7 +14,7 @@ from pathlib import Path
from .one_qubit_model import QubitModel, QubitModelMutliBath
from .two_qubit_model import TwoQubitModel
from .otto_cycle import OttoEngine
from collections.abc import Sequence, Iterator, Iterable
from collections.abc import Sequence, Iterator, Iterable, Callable
import shutil
import logging
import copy
@ -407,16 +407,23 @@ def cleanup(
def migrate_db_to_new_hashes(
data_path: str = "./.data", results_path: str = "./results"
data_path: str = "./.data", results_path: str = "./results", patch_fn: Optional[Callable] = None
):
"""
Recomputes all the hashes of the models in the database under
``data_path`` and updates the database.
``data_path`` and updates the database. The ``patch_fn`` receives
the database data and can return an arbitrarily modified version
of it from which the new hash is computed. This is useful if some
miss-configuration in the models has to be corrected.
"""
with model_db(data_path) as db:
for old_hash in list(db.keys()):
data = copy.deepcopy(db[old_hash])
if patch_fn:
data = patch_fn(data)
new_hash = data["model_config"].hexhash
del db[old_hash]
db[new_hash] = data