training/scripts/jsonify.py

from pathlib import Path
import pickle
import json

p = Path("training/artifacts/hierarchy/hierarchy_embed.pkl")

with open(p, "rb") as f:
    hierarchy = pickle.load(f)

hierarchy_json = {k: v.astype(int).tolist() for k, v in hierarchy.items()}

with open("training/artifacts/hierarchy/hierarchy_embed.json", "w") as f:
    json.dump(hierarchy_json, f)

print("done")