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

add migration routine

Browse source
This commit is contained in:
Valentin Boettcher 2022-04-11 14:49:02 +02:00
parent ae183ce045
commit 7ea10917c7
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
hiro_models/model_auxiliary.py
View file

@ -15,6 +15,7 @@ from .two_qubit_model import TwoQubitModel
from collections.abc import Sequence, Iterator, Iterable
import shutil
import logging
import copy
@contextmanager
@ -263,3 +264,18 @@ def cleanup(
if not preview:
del db[hash]
def migrate_db_to_new_hashes(data_path: str = "./.data"):
"""
Recomputes all the hashes of the models in the database under
``data_path`` and updates the database.
"""
with model_db(data_path) as db:
for old_hash in list(db.keys()):
data = copy.deepcopy(db[old_hash])
new_hash = data["model_config"].hexhash
del db[old_hash]
db[new_hash] = data