What’s your compute_dag that causes this problem? If it is a bug, you are welcome to contribute a patch.