def get_npz_path(checkpoint_path: str) -> str:
    """get the npz path for a specific checkpoint file"""
    pt_filename = checkpoint_path.split("/")[-1]
    model_id = pt_filename.split(".")[0]
    return "parity/npz-files/" + model_id + ".npz"