Move the graph compression back to caspo.learn and do the compression in main