Yes that's a good point. Causal DAGs add a layer of complexity that make this a more challenging solution.

I'm not very familiar with these frameworks though, so maybe it's pretty simple to add. Thanks for sharing the library!