Learning Causal Structure Distributions for Robust Planning