Migrate from GTO-based model registry to DVC-based model registry: move artifacts' annotations from `artifacts.yaml` to `dvc.yaml`

migrate.py

"""
Script that will
1. read your annotations from `artifacts.yaml`
2. update them to be compatible with the new DVC format
3. add them to `dvc.yaml`, creating it if needed
4. remove `artifacts.yaml`
"""
from ruamel.yaml import YAML

yaml = YAML(typ="safe", pure=True)
yaml.default_flow_style = False

# Read the YAML file
with open("artifacts.yaml", "r") as f:
    artifacts = yaml.load(f)

# Modify the format
for key, value in artifacts.items():
    if "virtual" in value:
        value.pop("virtual")
    if "description" in value:
        value["desc"] = value.pop("description")
    if "custom" in value:
        value["meta"] = value.pop("custom")

# Copy info from dvc.yaml
with open("dvc.yaml", "r") as f:
    dvcyaml = f.readlines()

# Write the modified data back to the same file
with open("dvc.yaml", "w") as f:
    yaml.dump({"artifacts": artifacts}, f)
    f.write("\n")
    f.writelines(dvcyaml)