import sys
from bedit_core.serialization import load, save
from bedit_core.bondgraph import causality_inference

def main(path: str):
    doc = load(path)
    for id, root in doc.root.items():
        doc.root[id] = causality_inference(doc.root[id])
    save(doc, path)

if __name__ == "__main__":
    main(sys.argv[1])